Add missing namespace

rapidsai · Jul 31, 2024 · 507777c · 507777c
1 parent 42bfd54
commit 507777c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/src/prims/key_store.cuh b/cpp/src/prims/key_store.cuh
@@ -323,7 +323,7 @@ class key_cuco_store_t {
       static_cast<size_t>(static_cast<double>(num_keys) / load_factor),
       static_cast<size_t>(num_keys) + 1);  // cuco::static_map requires at least one empty slot
 
-    auto stream_adapter = stream_allocator_adaptor(
+    auto stream_adapter = rmm::mr::stream_allocator_adaptor(
       rmm::mr::polymorphic_allocator<std::byte>(rmm::mr::get_current_device_resource()), stream);
     cuco_store_ =
       std::make_unique<cuco_set_type>(cuco_size,