From 9ced2434aabf3c1f96f6051be6c13bece6e59b68 Mon Sep 17 00:00:00 2001
From: Mark Harris <mharris@nvidia.com>
Date: Tue, 10 Sep 2024 20:42:50 +0000
Subject: [PATCH] set_current_device_resource() (not ref)

---
 cpp/bench/prims/neighbors/refine.cuh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cpp/bench/prims/neighbors/refine.cuh b/cpp/bench/prims/neighbors/refine.cuh
index c29349927a..b2d85a8085 100644
--- 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);