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

chore: bump sgl-kernel v0.1.7 (#6963)

Browse Source
This commit is contained in:
Yineng Zhang
2025-06-08 02:43:15 -07:00
committed by GitHub
parent 4740288303
commit 6c0a48282a
7 changed files with 13 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
sgl-kernel/build.sh
View File

@@ -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 \