From 9325207c6f63ed50eb773bbf5295527d6306e3ab Mon Sep 17 00:00:00 2001
From: Cedric Augonnet <caugonnet@nvidia.com>
Date: Thu, 12 Dec 2024 16:37:10 +0100
Subject: [PATCH] Fetch main branch of CCCL, using cudax

---
 cpp/cmake/thirdparty/cccl_override.json | 9 +++++++++
 cpp/cmake/thirdparty/get_cccl.cmake     | 7 +++++++
 2 files changed, 16 insertions(+)
 create mode 100644 cpp/cmake/thirdparty/cccl_override.json

diff --git a/cpp/cmake/thirdparty/cccl_override.json b/cpp/cmake/thirdparty/cccl_override.json
new file mode 100644
index 00000000000..0226e08a7bb
--- /dev/null
+++ b/cpp/cmake/thirdparty/cccl_override.json
@@ -0,0 +1,9 @@
+{
+    "packages": {
+      "cccl": {
+        "version": "2.8.0",
+        "git_url": "https://github.com/NVIDIA/cccl.git",
+        "git_tag": "main"
+      }
+    }
+}
diff --git a/cpp/cmake/thirdparty/get_cccl.cmake b/cpp/cmake/thirdparty/get_cccl.cmake
index 72b53d4c833..5acc0a60224 100644
--- a/cpp/cmake/thirdparty/get_cccl.cmake
+++ b/cpp/cmake/thirdparty/get_cccl.cmake
@@ -15,6 +15,13 @@
 # This function finds CCCL and sets any additional necessary environment variables.
 function(find_and_configure_cccl)
   include(${rapids-cmake-dir}/cpm/cccl.cmake)
+  include(${rapids-cmake-dir}/cpm/package_override.cmake)
+
+  rapids_cpm_package_override("${CMAKE_CURRENT_FUNCTION_LIST_DIR}/cccl_override.json")
+
+  # Enable cudax namespace install
+  set(CCCL_ENABLE_UNSTABLE ON)
+
   rapids_cpm_cccl(BUILD_EXPORT_SET cugraph-exports INSTALL_EXPORT_SET cugraph-exports)
 endfunction()