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

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

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
$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
$PIP_CMD list
python3 -c "import torch; print(torch.version.cuda)"
python3 -c "import cutlass; import cutlass.cute;"
# Prepare the CI runner (cleanup HuggingFace cache, etc.)
bash "${SCRIPT_DIR}/prepare_runner.sh"