File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change 28
28
#include < cuco/static_set.cuh>
29
29
30
30
#include < algorithm>
31
+ #include < cstddef>
31
32
#include < memory>
32
33
#include < optional>
33
34
#include < type_traits>
@@ -322,7 +323,7 @@ class key_cuco_store_t {
322
323
static_cast <size_t >(static_cast <double >(num_keys) / load_factor),
323
324
static_cast <size_t >(num_keys) + 1 ); // cuco::static_map requires at least one empty slot
324
325
325
- auto stream_adapter = rmm::mr::make_stream_allocator_adaptor (
326
+ auto stream_adapter = rmm::mr::stream_allocator_adaptor (
326
327
rmm::mr::polymorphic_allocator<std::byte>(rmm::mr::get_current_device_resource ()), stream);
327
328
cuco_store_ =
328
329
std::make_unique<cuco_set_type>(cuco_size,
Original file line number Diff line number Diff line change @@ -820,7 +820,7 @@ class kv_cuco_store_t {
820
820
static_cast <size_t >(static_cast <double >(num_keys) / load_factor),
821
821
static_cast <size_t >(num_keys) + 1 ); // cuco::static_map requires at least one empty slot
822
822
823
- auto stream_adapter = rmm::mr::make_stream_allocator_adaptor (
823
+ auto stream_adapter = rmm::mr::stream_allocator_adaptor (
824
824
rmm::mr::polymorphic_allocator<std::byte>(rmm::mr::get_current_device_resource ()), stream);
825
825
if constexpr (std::is_arithmetic_v<value_t >) {
826
826
cuco_store_ =
You can’t perform that action at this time.
0 commit comments