Not ready for _ref here yet.

rapidsai · Sep 3, 2024 · c7e91a0 · c7e91a0
1 parent 2131acb
commit c7e91a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cpp/bench/prims/common/benchmark.hpp b/cpp/bench/prims/common/benchmark.hpp
@@ -51,14 +51,14 @@ struct using_pool_memory_res {
 
  public:
   using_pool_memory_res(size_t initial_size, size_t max_size)
-    : orig_res_(raft::resource::get_current_device_resource_ref()),
+    : orig_res_(raft::resource::get_current_device_resource()),
       pool_res_(&cuda_res_, initial_size, max_size)
   {
     raft::resource::set_current_device_resource(&pool_res_);
   }
 
   using_pool_memory_res()
-    : orig_res_(raft::resource::get_current_device_resource_ref()),
+    : orig_res_(raft::resource::get_current_device_resource()),
       pool_res_(&cuda_res_, rmm::percent_of_free_device_memory(50))
   {
     raft::resource::set_current_device_resource(&pool_res_);