From 6b291eecfd993afc813a53478fbb1b6ec9a09f16 Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Tue, 5 Dec 2023 16:23:03 -0600
Subject: [PATCH] Update to rapids_cpm_cccl.

---
 cpp/CMakeLists.txt                                  | 10 +++++-----
 .../thirdparty/{get_thrust.cmake => get_cccl.cmake} | 13 ++++++-------
 2 files changed, 11 insertions(+), 12 deletions(-)
 rename cpp/cmake/thirdparty/{get_thrust.cmake => get_cccl.cmake} (71%)

diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt
index bccbc8c471..687af44fb3 100644
--- a/cpp/CMakeLists.txt
+++ b/cpp/CMakeLists.txt
@@ -177,8 +177,8 @@ include(cmake/modules/ConfigureCUDA.cmake)
 rapids_cpm_init()
 
 if(NOT BUILD_CPU_ONLY)
-  # thrust before rmm/cuco so we get the right version of thrust/cub
-  include(cmake/thirdparty/get_thrust.cmake)
+  # CCCL before rmm/cuco so we get the right version of CCCL
+  include(cmake/thirdparty/get_cccl.cmake)
   include(cmake/thirdparty/get_rmm.cmake)
   include(cmake/thirdparty/get_cutlass.cmake)
 
@@ -206,7 +206,7 @@ target_include_directories(
 
 if(NOT BUILD_CPU_ONLY)
   # Keep RAFT as lightweight as possible. Only CUDA libs and rmm should be used in global target.
-  target_link_libraries(raft INTERFACE rmm::rmm cuco::cuco nvidia::cutlass::cutlass raft::Thrust)
+  target_link_libraries(raft INTERFACE rmm::rmm cuco::cuco nvidia::cutlass::cutlass CCCL::Thrust)
 endif()
 
 target_compile_features(raft INTERFACE cxx_std_17 $<BUILD_INTERFACE:cuda_std_17>)
@@ -632,8 +632,8 @@ string(
   APPEND
   code_string
   [=[
-if(NOT TARGET raft::Thrust)
-  thrust_create_target(raft::Thrust FROM_OPTIONS)
+if(NOT TARGET CCCL::Thrust)
+  thrust_create_target(CCCL::Thrust FROM_OPTIONS)
 endif()
 ]=]
 )
diff --git a/cpp/cmake/thirdparty/get_thrust.cmake b/cpp/cmake/thirdparty/get_cccl.cmake
similarity index 71%
rename from cpp/cmake/thirdparty/get_thrust.cmake
rename to cpp/cmake/thirdparty/get_cccl.cmake
index 6e37aab40d..5c3ec76541 100644
--- a/cpp/cmake/thirdparty/get_thrust.cmake
+++ b/cpp/cmake/thirdparty/get_cccl.cmake
@@ -12,13 +12,12 @@
 # the License.
 # =============================================================================
 
-# Use CPM to find or clone thrust
-function(find_and_configure_thrust)
-        include(${rapids-cmake-dir}/cpm/thrust.cmake)
+# Use CPM to find or clone CCCL
+function(find_and_configure_cccl)
+        include(${rapids-cmake-dir}/cpm/cccl.cmake)
 
-        rapids_cpm_thrust( NAMESPACE raft
-                           BUILD_EXPORT_SET raft-exports
-                           INSTALL_EXPORT_SET raft-exports)
+        rapids_cpm_cccl( BUILD_EXPORT_SET raft-exports
+                         INSTALL_EXPORT_SET raft-exports)
 endfunction()
 
-find_and_configure_thrust()
+find_and_configure_cccl()