diff --git a/requirements-test-cuda.txt b/requirements-test-cuda.txt index 52a403aa9..d40f2221e 100644 --- a/requirements-test-cuda.txt +++ b/requirements-test-cuda.txt @@ -8,4 +8,4 @@ flash-attn==2.5.8 cuda-python==12.9.4 # CuTeDSL (CUTLASS Python DSL with CuTe support) -nvidia-cutlass-dsl>=4.3.1 +nvidia-cutlass-dsl==4.3.3