Upgrade cutlass to v3.3.0

Signed-off-by: Michael Tuttle <quic_mtuttle@quicinc.com>
quic · Dec 18, 2023 · a9e8fe9 · a9e8fe9
1 parent 87f8a87
commit a9e8fe9
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Jenkins/Dockerfile.torch-gpu b/Jenkins/Dockerfile.torch-gpu
@@ -273,9 +273,9 @@ RUN ln -s /usr/bin/run-clang-tidy-11.py /usr/bin/run-clang-tidy.py
 # Provide write permission to the python package folder in order to install AIMET in editable mode
 RUN chmod o+w /usr/local/lib/python3.8/dist-packages
 
-# Install cutlass 3.0
+# Install cutlass 3.3.0
 RUN cd /usr/local && \
-    git clone --depth 1 --branch v3.0.0 https://github.com/NVIDIA/cutlass.git && \
-    mv cutlass cutlass-3.0 && \
-    chown -R root:users cutlass-3.0 && \
-    ln -s cutlass-3.0 cutlass
+    git clone --depth 1 --branch v3.3.0 https://github.com/NVIDIA/cutlass.git && \
+    mv cutlass cutlass-3.3.0 && \
+    chown -R root:users cutlass-3.3.0 && \
+    ln -s cutlass-3.3.0 cutlass