Fix cutedsl ci error (#20309)

Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>
This commit is contained in:
Xiaoyu Zhang
2026-03-11 16:17:35 +08:00
committed by GitHub
co-authored by Baizhou Zhang
parent 456934fed5
commit 680d9d98e4
2 changed files with 5 additions and 2 deletions
+2 -2
View File
@@ -38,7 +38,7 @@ dependencies = [
"msgspec", "msgspec",
"ninja", "ninja",
"numpy", "numpy",
"nvidia-cutlass-dsl>=4.3.4", "nvidia-cutlass-dsl==4.3.5",
"nvidia-ml-py", "nvidia-ml-py",
"openai-harmony==0.0.4", "openai-harmony==0.0.4",
"openai==2.6.1", "openai==2.6.1",
@@ -59,7 +59,7 @@ dependencies = [
"scipy", "scipy",
"sentencepiece", "sentencepiece",
"setproctitle", "setproctitle",
"sgl-fa4==4.0.3", "sgl-fa4==4.0.4",
"sgl-kernel==0.3.21", "sgl-kernel==0.3.21",
"soundfile==0.13.1", "soundfile==0.13.1",
"tiktoken", "tiktoken",
+3
View File
@@ -328,6 +328,9 @@ fi
# Download flashinfer cubins if the local set is incomplete # Download flashinfer cubins if the local set is incomplete
bash "${SCRIPT_DIR}/ci_download_flashinfer_cubin.sh" 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
# 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)"