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

2026-03-11 13:37:33 -07:00
parent 2720ea2667
commit ab4b863546
2 changed files with 7 additions and 0 deletions
@@ -321,6 +321,11 @@ RUN --mount=type=cache,target=/root/.cache/pip \
    && rm -rf /root/.cargo /root/.rustup target dist ~/.cargo \
    && sed -i '/\.cargo\/env/d' /root/.profile /root/.bashrc 2>/dev/null || true

+# Pin nvidia-cutlass-dsl to 4.3.5
+RUN --mount=type=cache,target=/root/.cache/pip \
+   python3 -m pip uninstall nvidia-cutlass-dsl-libs-base; \
+   python3 -m pip install nvidia-cutlass-dsl==4.3.5 --force-reinstall;
+
 # Patching packages for CUDA 12/13 compatibility
 # TODO: Remove when torch version covers these packages
 RUN --mount=type=cache,target=/root/.cache/pip if [ "${CUDA_VERSION%%.*}" = "12" ]; then \