Check raft: faster matrix-vector-ops

Test cuml with changes in rapidsai/raft#401
achirkin · Dec 8, 2021 · 048066d · 048066d
1 parent f678c2c
commit 048066d
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
@@ -56,6 +56,6 @@ set(CUML_BRANCH_VERSION_raft "${CUML_VERSION_MAJOR}.${CUML_VERSION_MINOR}")
 # 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       achirkin
+                        PINNED_TAG enh-faster-linewise-ops
                         )