set_current_device_resource() (not ref)

rapidsai · Sep 10, 2024 · 9ced243 · 9ced243
1 parent 2fa415c
commit 9ced243
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/bench/prims/neighbors/refine.cuh b/cpp/bench/prims/neighbors/refine.cuh
@@ -59,7 +59,7 @@ class RefineAnn : public fixture {
     label_stream << data.p;
     state.SetLabel(label_stream.str());
 
-    auto old_mr = raft::resource::get_current_device_resource_ref();
+    auto old_mr = raft::resource::get_current_device_resource();
     rmm::mr::pool_memory_resource<rmm::mr::device_memory_resource> pool_mr(
       old_mr, rmm::percent_of_free_device_memory(50));
     raft::resource::set_current_device_resource(&pool_mr);