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

feat: support sgl-kernel cu129 (#8800)

Browse Source
This commit is contained in:
Yineng Zhang
2025-08-05 02:33:47 -07:00
committed by GitHub
parent 5e91fed1c5
commit 194561f27a
3 changed files with 87 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
sgl-kernel/build.sh
View File

@@ -21,7 +21,10 @@ else
BUILDER_NAME="pytorch/manylinux2_28-builder"
fi
if [ ${CUDA_VERSION} = "12.8" ]; then
if [ ${CUDA_VERSION} = "12.9" ]; then
DOCKER_IMAGE="${BUILDER_NAME}:cuda${CUDA_VERSION}"
TORCH_INSTALL="pip install --no-cache-dir torch==2.8.0 --index-url https://download.pytorch.org/whl/test/cu129"
elif [ ${CUDA_VERSION} = "12.8" ]; then
DOCKER_IMAGE="${BUILDER_NAME}:cuda${CUDA_VERSION}"
TORCH_INSTALL="pip install --no-cache-dir torch==2.8.0 --index-url https://download.pytorch.org/whl/test/cu128"
else