From 680d9d98e4683bfe3b63d25ae78dc0a7139f548c Mon Sep 17 00:00:00 2001
From: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>
Date: Wed, 11 Mar 2026 16:17:35 +0800
Subject: [PATCH] Fix cutedsl ci error (#20309)

Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>
---
 python/pyproject.toml                    | 4 ++--
 scripts/ci/cuda/ci_install_dependency.sh | 3 +++
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/python/pyproject.toml b/python/pyproject.toml
index 3b2bc9b93..e69ae38a5 100755
--- a/python/pyproject.toml
+++ b/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",
diff --git a/scripts/ci/cuda/ci_install_dependency.sh b/scripts/ci/cuda/ci_install_dependency.sh
index 527472fcd..c6fa958e9 100755
--- a/scripts/ci/cuda/ci_install_dependency.sh
+++ b/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)"