Make sure the logger is linked to the benchmarks

rapidsai · Nov 27, 2024 · aaadff5 · aaadff5
1 parent a0da5a9
commit aaadff5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cpp/bench/prims/CMakeLists.txt b/cpp/bench/prims/CMakeLists.txt
@@ -32,6 +32,7 @@ function(ConfigureBench)
     PRIVATE raft::raft
             raft_internal
             $<$<BOOL:${ConfigureBench_LIB}>:raft::compiled>
+            $<$<NOT:$<BOOL:${ConfigureBench_LIB}>>:bench_rmm_logger>
             ${RAFT_CTK_MATH_DEPENDENCIES}
             benchmark::benchmark
             Threads::Threads
@@ -73,6 +74,9 @@ function(ConfigureBench)
 
 endfunction()
 
+add_library(bench_rmm_logger OBJECT)
+target_link_libraries(bench_rmm_logger PRIVATE rmm::rmm_logger_impl)
+
 if(BUILD_PRIMS_BENCH)
   ConfigureBench(NAME CORE_BENCH PATH core/bitset.cu core/copy.cu main.cpp)