minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

fix ci by removing nvidia-cutlass-dsl-libs-base and force reinstall n… (#20380)

Browse Source
This commit is contained in:
Rain Jiang
2026-03-11 13:37:33 -07:00
committed by GitHub
parent 2720ea2667
commit ab4b863546
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+5
View File
@@ -321,6 +321,11 @@ RUN --mount=type=cache,target=/root/.cache/pip \
&& rm -rf /root/.cargo /root/.rustup target dist ~/.cargo \ && rm -rf /root/.cargo /root/.rustup target dist ~/.cargo \
&& sed -i '/\.cargo\/env/d' /root/.profile /root/.bashrc 2>/dev/null || true && sed -i '/\.cargo\/env/d' /root/.profile /root/.bashrc 2>/dev/null || true
# Pin nvidia-cutlass-dsl to 4.3.5
RUN --mount=type=cache,target=/root/.cache/pip \
python3 -m pip uninstall nvidia-cutlass-dsl-libs-base; \
python3 -m pip install nvidia-cutlass-dsl==4.3.5 --force-reinstall;
# Patching packages for CUDA 12/13 compatibility # Patching packages for CUDA 12/13 compatibility
# TODO: Remove when torch version covers these packages # TODO: Remove when torch version covers these packages
RUN --mount=type=cache,target=/root/.cache/pip if [ "${CUDA_VERSION%%.*}" = "12" ]; then \ RUN --mount=type=cache,target=/root/.cache/pip if [ "${CUDA_VERSION%%.*}" = "12" ]; then \
scripts/ci/cuda/ci_install_dependency.sh
+2
View File
@@ -330,10 +330,12 @@ bash "${SCRIPT_DIR}/ci_download_flashinfer_cubin.sh"
# Clean nvidia-cutlass-dsl-libs-base for cutedsl lower than 0.4.4 # Clean nvidia-cutlass-dsl-libs-base for cutedsl lower than 0.4.4
$PIP_UNINSTALL_CMD nvidia-cutlass-dsl-libs-base $PIP_UNINSTALL_SUFFIX || true $PIP_UNINSTALL_CMD nvidia-cutlass-dsl-libs-base $PIP_UNINSTALL_SUFFIX || true
$PIP_CMD install nvidia-cutlass-dsl==4.3.5 --force-reinstall $PIP_INSTALL_SUFFIX || true
# Show current packages # Show current packages
$PIP_CMD list $PIP_CMD list
python3 -c "import torch; print(torch.version.cuda)" python3 -c "import torch; print(torch.version.cuda)"
python3 -c "import cutlass; import cutlass.cute;"
# Prepare the CI runner (cleanup HuggingFace cache, etc.) # Prepare the CI runner (cleanup HuggingFace cache, etc.)
bash "${SCRIPT_DIR}/prepare_runner.sh" bash "${SCRIPT_DIR}/prepare_runner.sh"