Test changes in raft device atomics

rapidsai · Jul 8, 2022 · b41b0e9 · b41b0e9
1 parent c8aebc3
commit b41b0e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cpp/cmake/thirdparty/get_raft.cmake b/cpp/cmake/thirdparty/get_raft.cmake
@@ -78,8 +78,8 @@ endfunction()
 # To use a different RAFT locally, set the CMake variable
 # CPM_raft_SOURCE=/path/to/local/raft
 find_and_configure_raft(VERSION          ${CUML_MIN_VERSION_raft}
-                        FORK             rapidsai
-                        PINNED_TAG       branch-${CUML_BRANCH_VERSION_raft}
+                        FORK             ahendriksen
+                        PINNED_TAG       fea-add-warp-aggregated-atomic-increment
 
                         # When PINNED_TAG above doesn't match cuml,
                         # force local raft clone in build directory