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

chore: use torch 2.6 for sgl-kernel build (#5898)

parent e3a53044
......@@ -13,7 +13,7 @@ if [ ${CUDA_VERSION} = "12.8" ]; then
TORCH_INSTALL="pip install --no-cache-dir --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION//.}"
else
DOCKER_IMAGE="pytorch/manylinux-builder:cuda${CUDA_VERSION}"
TORCH_INSTALL="pip install --no-cache-dir torch==2.5.1 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}"
TORCH_INSTALL="pip install --no-cache-dir torch==2.6.0 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}"
fi
docker run --rm \
......
[build-system]
requires = [
"scikit-build-core>=0.10",
"torch>=2.5.1",
"torch>=2.6.0",
"wheel",
]
build-backend = "scikit_build_core.build"
......
......@@ -2,7 +2,7 @@
requires = [
"setuptools>=75.0",
"scikit-build-core>=0.10",
"torch>=2.5.1",
"torch>=2.6.0",
"wheel",
]
build-backend = "setuptools.build_meta"
......
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