Merge branch 'branch-24.10' into dgl-examples

rapidsai · Aug 5, 2024 · 6c9a5bf · 6c9a5bf
2 parents 139b3d6 + 8f86c82
commit 6c9a5bf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cpp/src/prims/key_store.cuh b/cpp/src/prims/key_store.cuh
@@ -28,6 +28,7 @@
 #include <cuco/static_set.cuh>
 
 #include <algorithm>
+#include <cstddef>
 #include <memory>
 #include <optional>
 #include <type_traits>
@@ -322,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 = rmm::mr::make_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,

diff --git a/cpp/src/prims/kv_store.cuh b/cpp/src/prims/kv_store.cuh
@@ -820,7 +820,7 @@ class kv_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 = rmm::mr::make_stream_allocator_adaptor(
+    auto stream_adapter = rmm::mr::stream_allocator_adaptor(
       rmm::mr::polymorphic_allocator<std::byte>(rmm::mr::get_current_device_resource()), stream);
     if constexpr (std::is_arithmetic_v<value_t>) {
       cuco_store_ =