Use an override to test in static compilation settings

rapidsai · Nov 27, 2024 · e991c5c · e991c5c
1 parent 12b415a
commit e991c5c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/cpp/cmake/thirdparty/get_rmm.cmake b/cpp/cmake/thirdparty/get_rmm.cmake
@@ -1,5 +1,5 @@
 # =============================================================================
-# Copyright (c) 2020-2021, NVIDIA CORPORATION.
+# Copyright (c) 2020-2024, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 # in compliance with the License. You may obtain a copy of the License at
@@ -15,6 +15,10 @@
 # This function finds rmm and sets any additional necessary environment variables.
 function(find_and_configure_rmm)
   include(${rapids-cmake-dir}/cpm/rmm.cmake)
+  include(${rapids-cmake-dir}/cpm/package_override.cmake)
+
+  set(cudf_patch_dir "${CMAKE_CURRENT_FUNCTION_LIST_DIR}/patches")
+  rapids_cpm_package_override("${cudf_patch_dir}/rmm_override.json")
 
   # Find or install RMM
   rapids_cpm_rmm(BUILD_EXPORT_SET cudf-exports INSTALL_EXPORT_SET cudf-exports)

diff --git a/cpp/cmake/thirdparty/patches/rmm_override.json b/cpp/cmake/thirdparty/patches/rmm_override.json
@@ -0,0 +1,9 @@
+{
+  "packages": {
+    "rmm": {
+      "version": "25.02",
+      "git_url": "https://github.com/vyasr/rmm.git",
+      "git_tag": "fix/logger_issues"
+    }
+  }
+}