Adapt to rmm logger changes (#6147)

This PR adapts to breaking changes in rmm in rapidsai/rmm#1722. --------- Co-authored-by: Bradley Dice <bdice@bradleydice.com>
rapidsai · Dec 2, 2024 · 6f8a189 · 6f8a189
1 parent ae9c1ea
commit 6f8a189
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt
@@ -626,9 +626,10 @@ if(BUILD_CUML_CPP_LIBRARY)
   )
 
   target_link_libraries(${CUML_CPP_TARGET}
-    PUBLIC  rmm::rmm ${CUVS_LIB}
+    PUBLIC  rmm::rmm rmm::rmm_logger ${CUVS_LIB}
             ${_cuml_cpp_public_libs}
     PRIVATE ${_cuml_cpp_private_libs}
+                   rmm::rmm_logger_impl
   )
 
   # If we export the libdmlc symbols, they can lead to weird crashes with other

diff --git a/cpp/bench/CMakeLists.txt b/cpp/bench/CMakeLists.txt
@@ -50,6 +50,7 @@ if(BUILD_CUML_BENCH)
       benchmark::benchmark
       ${TREELITE_LIBS}
       raft::raft
+      rmm::rmm_logger_impl
   )
 
   target_include_directories(${CUML_CPP_BENCH_TARGET}

diff --git a/cpp/test/CMakeLists.txt b/cpp/test/CMakeLists.txt
@@ -56,6 +56,8 @@ function(ConfigureTest)
     CUDA::cusparse${_ctk_static_suffix}
     $<$<BOOL:${LINK_CUFFT}>:CUDA::cufft${_ctk_static_suffix_cufft}>
     rmm::rmm
+    rmm::rmm_logger
+    rmm::rmm_logger_impl
     raft::raft
     GTest::gtest
     GTest::gtest_main