chenchenghao/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 11 Packages Projects Releases Wiki Activity

[CI] Fix cutlass import error: restore nvidia-cutlass-dsl force-reinstall (#21182)

Browse Source 
Co-authored-by: Alison Shao <alison.shao@mac.lan>
This commit is contained in:
Alison Shao
2026-03-22 22:41:01 -07:00
committed by GitHub
parent 7050011dee
commit 44db0c59cf
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/ci/cuda/ci_install_dependency.sh
View File

@@ -340,6 +340,11 @@ fi
mark_step_done "Fix other dependencies"
# Force reinstall nvidia-cutlass-dsl to ensure the .pth file exists.
# The Docker image ships nvidia-cutlass-dsl-libs-base 4.3.5; upgrading to 4.4.2
# can delete the .pth file without reliably recreating it (pip race condition).
$PIP_CMD install "nvidia-cutlass-dsl>=4.4.1" "nvidia-cutlass-dsl-libs-base>=4.4.1" --no-deps --force-reinstall $PIP_INSTALL_SUFFIX || true
# ------------------------------------------------------------------------------
# Prepare runner
# ------------------------------------------------------------------------------