diff --git a/.github/workflows/pr-test-sgl-kernel.yml b/.github/workflows/pr-test-sgl-kernel.yml index 9fd20bd9a..5081ebedc 100644 --- a/.github/workflows/pr-test-sgl-kernel.yml +++ b/.github/workflows/pr-test-sgl-kernel.yml @@ -36,8 +36,6 @@ jobs: strategy: matrix: include: - - python-version: '3.9' - cuda-version: '11.8' - python-version: '3.9' cuda-version: '12.4' - python-version: '3.9' @@ -88,7 +86,7 @@ jobs: - name: Install run: | 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 install sgl-kernel/dist/*whl --force-reinstall --no-deps pip3 list | grep sgl-kernel @@ -120,6 +118,7 @@ jobs: - name: Install run: | bash scripts/ci_install_dependency.sh + pip3 install torch==2.7.1 torchvision pip3 uninstall sgl-kernel -y || true pip3 install sgl-kernel/dist/*whl --force-reinstall --no-deps pip3 list | grep sgl-kernel diff --git a/docker/Dockerfile.blackwell b/docker/Dockerfile.blackwell index 3cc1a0f70..4f3053971 100644 --- a/docker/Dockerfile.blackwell +++ b/docker/Dockerfile.blackwell @@ -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 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 RUN git clone --depth=1 https://github.com/sgl-project/sglang.git \ diff --git a/sgl-kernel/build.sh b/sgl-kernel/build.sh index 70a982ab2..d11027157 100755 --- a/sgl-kernel/build.sh +++ b/sgl-kernel/build.sh @@ -17,19 +17,15 @@ if [ ${ARCH} = "aarch64" ]; then BUILDER_NAME="pytorch/manylinuxaarch64-builder" else LIBCUDA_ARCH=${ARCH} - if [ ${CUDA_VERSION} = "12.8" ]; then - BUILDER_NAME="pytorch/manylinux2_28-builder" - else - BUILDER_NAME="pytorch/manylinux-builder" - fi + BUILDER_NAME="pytorch/manylinux2_28-builder" fi if [ ${CUDA_VERSION} = "12.8" ]; then 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 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 docker run --rm \ diff --git a/sgl-kernel/pyproject.toml b/sgl-kernel/pyproject.toml index 3d65dc68f..ef059207c 100644 --- a/sgl-kernel/pyproject.toml +++ b/sgl-kernel/pyproject.toml @@ -1,14 +1,14 @@ [build-system] requires = [ "scikit-build-core>=0.10", - "torch>=2.6.0", + "torch>=2.7.1", "wheel", ] build-backend = "scikit_build_core.build" [project] name = "sgl-kernel" -version = "0.1.6.post1" +version = "0.1.7" description = "Kernel Library for SGLang" readme = "README.md" requires-python = ">=3.9" diff --git a/sgl-kernel/pyproject_cpu.toml b/sgl-kernel/pyproject_cpu.toml index fae7b6cb5..a4482caee 100644 --- a/sgl-kernel/pyproject_cpu.toml +++ b/sgl-kernel/pyproject_cpu.toml @@ -1,14 +1,14 @@ [build-system] requires = [ "scikit-build-core>=0.10", - "torch>=2.6.0", + "torch>=2.7.1", "wheel", ] build-backend = "scikit_build_core.build" [project] name = "sgl-kernel" -version = "0.1.6.post1" +version = "0.1.7" description = "Kernel Library for SGLang" readme = "README.md" requires-python = ">=3.9" diff --git a/sgl-kernel/pyproject_rocm.toml b/sgl-kernel/pyproject_rocm.toml index 3e600a290..34abf9f92 100644 --- a/sgl-kernel/pyproject_rocm.toml +++ b/sgl-kernel/pyproject_rocm.toml @@ -2,14 +2,14 @@ requires = [ "setuptools>=75.0", "scikit-build-core>=0.10", - "torch>=2.6.0", + "torch>=2.7.1", "wheel", ] build-backend = "setuptools.build_meta" [project] name = "sgl-kernel" -version = "0.1.6.post1" +version = "0.1.7" description = "Kernel Library for SGLang" readme = "README.md" requires-python = ">=3.9" diff --git a/sgl-kernel/python/sgl_kernel/version.py b/sgl-kernel/python/sgl_kernel/version.py index 81c5ee2a4..f1380eede 100644 --- a/sgl-kernel/python/sgl_kernel/version.py +++ b/sgl-kernel/python/sgl_kernel/version.py @@ -1 +1 @@ -__version__ = "0.1.6.post1" +__version__ = "0.1.7"