Fix resource_ref init

rapidsai · Sep 10, 2024 · 44d9b11 · 44d9b11
1 parent dfa661e
commit 44d9b11
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/cpp/bench/prims/neighbors/knn.cuh b/cpp/bench/prims/neighbors/knn.cuh
@@ -91,13 +91,9 @@ inline auto operator<<(std::ostream& os, const Scope& s) -> std::ostream&
 struct device_resource {
  public:
   explicit device_resource(bool managed)
-    : managed_(managed ? std::make_shared<rmm::mr::managed_memory_resource>() : nullptr)
+    : managed_(managed ? std::make_shared<rmm::mr::managed_memory_resource>() : nullptr),
+      res_(managed ? managed.get() : raft::resource::get_current_device_resource_ref())
   {
-    if (managed) {
-      res_ = managed.get();
-    } else {
-      res_ = raft::resource::get_current_device_resource_ref();
-    }
   }
 
   ~device_resource() {}