Fix cutedsl ci error (#20309)

Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>

python/pyproject.toml

@@ -38,7 +38,7 @@ dependencies = [
   "msgspec",
   "ninja",
   "numpy",
-  "nvidia-cutlass-dsl>=4.3.4",
+  "nvidia-cutlass-dsl==4.3.5",
   "nvidia-ml-py",
   "openai-harmony==0.0.4",
   "openai==2.6.1",
@@ -59,7 +59,7 @@ dependencies = [
   "scipy",
   "sentencepiece",
   "setproctitle",
-  "sgl-fa4==4.0.3",
+  "sgl-fa4==4.0.4",
   "sgl-kernel==0.3.21",
   "soundfile==0.13.1",
   "tiktoken",

scripts/ci/cuda/ci_install_dependency.sh

@@ -328,6 +328,9 @@ fi
 # Download flashinfer cubins if the local set is incomplete
 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
 $PIP_CMD list
 python3 -c "import torch; print(torch.version.cuda)"