chore: bump sgl-kernel 0.3.0 with torch 2.8.0 (#8718)

This commit is contained in:
Yineng Zhang
2025-08-03 02:31:50 -07:00
committed by GitHub
parent 6f9baf1002
commit 5ce5093b97
6 changed files with 10 additions and 10 deletions

View File

@@ -23,10 +23,10 @@ fi
if [ ${CUDA_VERSION} = "12.8" ]; then
DOCKER_IMAGE="${BUILDER_NAME}:cuda${CUDA_VERSION}"
TORCH_INSTALL="pip install --no-cache-dir torch==2.7.1 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}"
TORCH_INSTALL="pip install --no-cache-dir torch==2.8.0 --index-url https://download.pytorch.org/whl/test/cu128"
else
DOCKER_IMAGE="${BUILDER_NAME}:cuda${CUDA_VERSION}"
TORCH_INSTALL="pip install --no-cache-dir torch==2.7.1"
TORCH_INSTALL="pip install --no-cache-dir torch==2.8.0 --index-url https://download.pytorch.org/whl/test/cu126"
fi
docker run --rm \