Skip to content

Commit a043f7f

Browse files
authored
chore: use torch 2.6 for sgl-kernel build (#5898)
1 parent e3a5304 commit a043f7f

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

sgl-kernel/build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ if [ ${CUDA_VERSION} = "12.8" ]; then
1313
TORCH_INSTALL="pip install --no-cache-dir --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION//.}"
1414
else
1515
DOCKER_IMAGE="pytorch/manylinux-builder:cuda${CUDA_VERSION}"
16-
TORCH_INSTALL="pip install --no-cache-dir torch==2.5.1 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}"
16+
TORCH_INSTALL="pip install --no-cache-dir torch==2.6.0 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}"
1717
fi
1818

1919
docker run --rm \

sgl-kernel/pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
[build-system]
22
requires = [
33
"scikit-build-core>=0.10",
4-
"torch>=2.5.1",
4+
"torch>=2.6.0",
55
"wheel",
66
]
77
build-backend = "scikit_build_core.build"

sgl-kernel/pyproject_rocm.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
requires = [
33
"setuptools>=75.0",
44
"scikit-build-core>=0.10",
5-
"torch>=2.5.1",
5+
"torch>=2.6.0",
66
"wheel",
77
]
88
build-backend = "setuptools.build_meta"

0 commit comments

Comments
 (0)