From 0261c4aff784389d9d50a0ae00cbeb8d4570981f Mon Sep 17 00:00:00 2001 From: Baizhou Zhang Date: Tue, 16 Dec 2025 12:11:56 -0800 Subject: [PATCH] [misc] Upgrade cutedsl to 4.3.1 (#14857) --- docker/Dockerfile | 3 --- python/pyproject.toml | 2 +- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 63da4f47d..02975e66b 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -301,16 +301,13 @@ RUN --mount=type=cache,target=/root/.cache/pip \ # Patching packages for CUDA 12/13 compatibility # TODO: Remove when torch version covers these packages -# TODO: Move cutlass-dsl to pyproject.toml after drivers on CI runners are updated RUN --mount=type=cache,target=/root/.cache/pip if [ "${CUDA_VERSION%%.*}" = "12" ]; then \ python3 -m pip install nvidia-nccl-cu12==2.28.3 --force-reinstall --no-deps ; \ python3 -m pip install nvidia-cudnn-cu12==9.16.0.29 --force-reinstall --no-deps ; \ - python3 -m pip install nvidia-cutlass-dsl==4.3.0 --force-reinstall --no-deps ; \ elif [ "${CUDA_VERSION%%.*}" = "13" ]; then \ python3 -m pip install nvidia-nccl-cu13==2.28.3 --force-reinstall --no-deps ; \ python3 -m pip install nvidia-cublas==13.1.0.3 --force-reinstall --no-deps ; \ python3 -m pip install nixl-cu13 ; \ - python3 -m pip install nvidia-cutlass-dsl==4.3.1 --force-reinstall --no-deps ; \ fi # Install development tools diff --git a/python/pyproject.toml b/python/pyproject.toml index 198792559..aef4f7c74 100755 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -38,7 +38,7 @@ dependencies = [ "msgspec", "ninja", "numpy", - "nvidia-cutlass-dsl==4.2.1", + "nvidia-cutlass-dsl==4.3.1", "nvidia-ml-py", "openai-harmony==0.0.4", "openai==2.6.1",