From 44db0c59cfb23b77c041d4fb833550635f801016 Mon Sep 17 00:00:00 2001 From: Alison Shao <54658187+alisonshao@users.noreply.github.com> Date: Sun, 22 Mar 2026 22:41:01 -0700 Subject: [PATCH] [CI] Fix cutlass import error: restore nvidia-cutlass-dsl force-reinstall (#21182) Co-authored-by: Alison Shao --- scripts/ci/cuda/ci_install_dependency.sh | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/scripts/ci/cuda/ci_install_dependency.sh b/scripts/ci/cuda/ci_install_dependency.sh index 53cfc5e90..dfaf0152e 100755 --- a/scripts/ci/cuda/ci_install_dependency.sh +++ b/scripts/ci/cuda/ci_install_dependency.sh @@ -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 # ------------------------------------------------------------------------------