From 7d5fd87bd8a99e12950a49e6b892a9e8df2cc562 Mon Sep 17 00:00:00 2001
From: Robert Maynard <rmaynard@nvidia.com>
Date: Thu, 9 Jan 2025 11:27:11 -0500
Subject: [PATCH] Adapt to rmm 25.02 logger changes

---
 CMakeLists.txt           | 1 -
 tests/cpp/CMakeLists.txt | 2 ++
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 761aef49d4dc..03336cd28270 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -329,7 +329,6 @@ if(PLUGIN_RMM)
   list(REMOVE_ITEM rmm_link_libs CUDA::cudart)
   list(APPEND rmm_link_libs CUDA::cudart_static)
   set_target_properties(rmm::rmm PROPERTIES INTERFACE_LINK_LIBRARIES "${rmm_link_libs}")
-  get_target_property(rmm_link_libs rmm::rmm INTERFACE_LINK_LIBRARIES)
 endif()
 
 if(PLUGIN_SYCL)
diff --git a/tests/cpp/CMakeLists.txt b/tests/cpp/CMakeLists.txt
index deed08165bc2..580fd560ed81 100644
--- a/tests/cpp/CMakeLists.txt
+++ b/tests/cpp/CMakeLists.txt
@@ -75,6 +75,8 @@ target_include_directories(testxgboost
   ${xgboost_SOURCE_DIR}/dmlc-core/include)
 target_link_libraries(testxgboost
   PRIVATE
+  $<TARGET_NAME_IF_EXISTS:rmm::rmm_logger>
+  $<TARGET_NAME_IF_EXISTS:rmm::rmm_logger_impl>
   GTest::gtest GTest::gmock)
 
 set_output_directory(testxgboost ${xgboost_BINARY_DIR})