Upgrade to latest cutlass version

rapidsai · Nov 20, 2024 · 1a75aad · 1a75aad
1 parent d8dc72c
commit 1a75aad
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cpp/cmake/thirdparty/get_cutlass.cmake b/cpp/cmake/thirdparty/get_cutlass.cmake
@@ -79,14 +79,15 @@ function(find_and_configure_cutlass)
   )
 endfunction()
 
+set(_cutlass_version 3.5.1)
 if(NOT RAFT_CUTLASS_GIT_TAG)
-  set(RAFT_CUTLASS_GIT_TAG v2.10.0)
+  set(RAFT_CUTLASS_GIT_TAG "v${_cutlass_version}")
 endif()
 
 if(NOT RAFT_CUTLASS_GIT_REPOSITORY)
   set(RAFT_CUTLASS_GIT_REPOSITORY https://github.com/NVIDIA/cutlass.git)
 endif()
 
 find_and_configure_cutlass(
-  VERSION 2.10.0 REPOSITORY ${RAFT_CUTLASS_GIT_REPOSITORY} PINNED_TAG ${RAFT_CUTLASS_GIT_TAG}
+  VERSION ${_cutlass_version} REPOSITORY ${RAFT_CUTLASS_GIT_REPOSITORY} PINNED_TAG ${RAFT_CUTLASS_GIT_TAG}
 )