Unverified Commit 6c0a4828 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

chore: bump sgl-kernel v0.1.7 (#6963)

parent 47402883
...@@ -36,8 +36,6 @@ jobs: ...@@ -36,8 +36,6 @@ jobs:
strategy: strategy:
matrix: matrix:
include: include:
- python-version: '3.9'
cuda-version: '11.8'
- python-version: '3.9' - python-version: '3.9'
cuda-version: '12.4' cuda-version: '12.4'
- python-version: '3.9' - python-version: '3.9'
...@@ -88,7 +86,7 @@ jobs: ...@@ -88,7 +86,7 @@ jobs:
- name: Install - name: Install
run: | run: |
bash scripts/ci_install_dependency.sh bash scripts/ci_install_dependency.sh
pip3 install torch==2.6.0 torchvision && pip3 install pytest pip3 install torch==2.7.1 torchvision && pip3 install pytest
pip3 uninstall sgl-kernel -y || true pip3 uninstall sgl-kernel -y || true
pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps
pip3 list | grep sgl-kernel pip3 list | grep sgl-kernel
...@@ -120,6 +118,7 @@ jobs: ...@@ -120,6 +118,7 @@ jobs:
- name: Install - name: Install
run: | run: |
bash scripts/ci_install_dependency.sh bash scripts/ci_install_dependency.sh
pip3 install torch==2.7.1 torchvision
pip3 uninstall sgl-kernel -y || true pip3 uninstall sgl-kernel -y || true
pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps
pip3 list | grep sgl-kernel pip3 list | grep sgl-kernel
......
...@@ -20,7 +20,7 @@ RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \ ...@@ -20,7 +20,7 @@ RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \
RUN pip3 install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128 --break-system-packages RUN pip3 install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128 --break-system-packages
RUN pip3 install https://github.com/sgl-project/whl/releases/download/v0.1.6.post1/sgl_kernel-0.1.6.post1+cu128-cp39-abi3-manylinux2014_x86_64.whl --break-system-packages \ RUN pip3 install https://github.com/sgl-project/whl/releases/download/v0.1.7/sgl_kernel-0.1.7+cu128-cp39-abi3-manylinux2014_x86_64.whl --break-system-packages \
&& pip3 install setuptools==75.0.0 wheel scikit-build-core --break-system-packages && pip3 install setuptools==75.0.0 wheel scikit-build-core --break-system-packages
RUN git clone --depth=1 https://github.com/sgl-project/sglang.git \ RUN git clone --depth=1 https://github.com/sgl-project/sglang.git \
......
...@@ -17,19 +17,15 @@ if [ ${ARCH} = "aarch64" ]; then ...@@ -17,19 +17,15 @@ if [ ${ARCH} = "aarch64" ]; then
BUILDER_NAME="pytorch/manylinuxaarch64-builder" BUILDER_NAME="pytorch/manylinuxaarch64-builder"
else else
LIBCUDA_ARCH=${ARCH} LIBCUDA_ARCH=${ARCH}
if [ ${CUDA_VERSION} = "12.8" ]; then BUILDER_NAME="pytorch/manylinux2_28-builder"
BUILDER_NAME="pytorch/manylinux2_28-builder"
else
BUILDER_NAME="pytorch/manylinux-builder"
fi
fi fi
if [ ${CUDA_VERSION} = "12.8" ]; then if [ ${CUDA_VERSION} = "12.8" ]; then
DOCKER_IMAGE="${BUILDER_NAME}:cuda${CUDA_VERSION}" DOCKER_IMAGE="${BUILDER_NAME}:cuda${CUDA_VERSION}"
TORCH_INSTALL="pip install --no-cache-dir torch==2.7.0 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}" TORCH_INSTALL="pip install --no-cache-dir torch==2.7.1 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}"
else else
DOCKER_IMAGE="${BUILDER_NAME}:cuda${CUDA_VERSION}" DOCKER_IMAGE="${BUILDER_NAME}:cuda${CUDA_VERSION}"
TORCH_INSTALL="pip install --no-cache-dir torch==2.6.0 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}" TORCH_INSTALL="pip install --no-cache-dir torch==2.7.1"
fi fi
docker run --rm \ docker run --rm \
......
[build-system] [build-system]
requires = [ requires = [
"scikit-build-core>=0.10", "scikit-build-core>=0.10",
"torch>=2.6.0", "torch>=2.7.1",
"wheel", "wheel",
] ]
build-backend = "scikit_build_core.build" build-backend = "scikit_build_core.build"
[project] [project]
name = "sgl-kernel" name = "sgl-kernel"
version = "0.1.6.post1" version = "0.1.7"
description = "Kernel Library for SGLang" description = "Kernel Library for SGLang"
readme = "README.md" readme = "README.md"
requires-python = ">=3.9" requires-python = ">=3.9"
......
[build-system] [build-system]
requires = [ requires = [
"scikit-build-core>=0.10", "scikit-build-core>=0.10",
"torch>=2.6.0", "torch>=2.7.1",
"wheel", "wheel",
] ]
build-backend = "scikit_build_core.build" build-backend = "scikit_build_core.build"
[project] [project]
name = "sgl-kernel" name = "sgl-kernel"
version = "0.1.6.post1" version = "0.1.7"
description = "Kernel Library for SGLang" description = "Kernel Library for SGLang"
readme = "README.md" readme = "README.md"
requires-python = ">=3.9" requires-python = ">=3.9"
......
...@@ -2,14 +2,14 @@ ...@@ -2,14 +2,14 @@
requires = [ requires = [
"setuptools>=75.0", "setuptools>=75.0",
"scikit-build-core>=0.10", "scikit-build-core>=0.10",
"torch>=2.6.0", "torch>=2.7.1",
"wheel", "wheel",
] ]
build-backend = "setuptools.build_meta" build-backend = "setuptools.build_meta"
[project] [project]
name = "sgl-kernel" name = "sgl-kernel"
version = "0.1.6.post1" version = "0.1.7"
description = "Kernel Library for SGLang" description = "Kernel Library for SGLang"
readme = "README.md" readme = "README.md"
requires-python = ">=3.9" requires-python = ">=3.9"
......
__version__ = "0.1.6.post1" __version__ = "0.1.7"
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment