EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Yineng Zhang
2025-04-29 17:51:18 -07:00
committed by GitHub
parent e3a5304475
commit a043f7f2ab
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sgl-kernel/build.sh
View File

@@ -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//.}" TORCH_INSTALL="pip install --no-cache-dir --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION//.}"
else else
DOCKER_IMAGE="pytorch/manylinux-builder:cuda${CUDA_VERSION}" 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 fi
docker run --rm \ docker run --rm \

2
sgl-kernel/pyproject.toml
View File

@@ -1,7 +1,7 @@
[build-system] [build-system]
requires = [ requires = [
"scikit-build-core>=0.10", "scikit-build-core>=0.10",
"torch>=2.5.1", "torch>=2.6.0",
"wheel", "wheel",
] ]
build-backend = "scikit_build_core.build" build-backend = "scikit_build_core.build"

2
sgl-kernel/pyproject_rocm.toml
View File

@@ -2,7 +2,7 @@
requires = [ requires = [
"setuptools>=75.0", "setuptools>=75.0",
"scikit-build-core>=0.10", "scikit-build-core>=0.10",
"torch>=2.5.1", "torch>=2.6.0",
"wheel", "wheel",
] ]
build-backend = "setuptools.build_meta" build-backend = "setuptools.build_meta"