Skip to content
Merged
Show file tree
Hide file tree
Changes from 13 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions cpp/benchmarks/bitmask/bitmask_and.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ void BM_segmented_bitmask_and(nvbench::state& state)

auto [segments, masks, mask_pointers, mask_begin_bits, data_bytes] = setup_masks(state);

state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
state.add_element_count(data_bytes, "input size");
state.template add_global_memory_reads<nvbench::int8_t>(data_bytes);
auto const mem_stats_logger = cudf::memory_stats_logger();
Expand All @@ -96,7 +96,7 @@ void BM_multi_segment_bitmask_and(nvbench::state& state)

auto [segments, masks, mask_pointers, mask_begin_bits, data_bytes] = setup_masks(state);

state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
state.add_element_count(data_bytes, "input size");
state.template add_global_memory_reads<nvbench::int8_t>(data_bytes);
auto const mem_stats_logger = cudf::memory_stats_logger();
Expand Down
10 changes: 5 additions & 5 deletions cpp/benchmarks/bitmask/set_null_mask.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -61,7 +61,7 @@ void BM_setnullmask(nvbench::state& state)
rmm::device_buffer mask = cudf::create_null_mask(mask_size, cudf::mask_state::UNINITIALIZED);
auto begin = 0, end = mask_size;

state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
auto const mem_stats_logger = cudf::memory_stats_logger();

state.exec(nvbench::exec_tag::sync | nvbench::exec_tag::timer,
Expand Down Expand Up @@ -90,7 +90,7 @@ void BM_setnullmask_unsafe_bulk(nvbench::state& state)
auto [begin_bits, end_bits, valids, masks, masks_ptr] =
generate_test_data(num_masks, mask_size, use_variable_mask_sizes);

state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
auto const mem_stats_logger = cudf::memory_stats_logger();

state.exec(nvbench::exec_tag::sync | nvbench::exec_tag::timer,
Expand Down Expand Up @@ -118,7 +118,7 @@ void BM_setnullmask_safe_bulk(nvbench::state& state)
auto [begin_bits, end_bits, valids, masks, masks_ptr] =
generate_test_data(num_masks, mask_size, use_variable_mask_sizes);

state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
auto const mem_stats_logger = cudf::memory_stats_logger();

state.exec(nvbench::exec_tag::sync | nvbench::exec_tag::timer,
Expand Down Expand Up @@ -146,7 +146,7 @@ void BM_setnullmask_loop(nvbench::state& state)
auto [begin_bits, end_bits, valids, masks, masks_ptr] =
generate_test_data(num_masks, mask_size, use_variable_mask_sizes);

state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
auto const mem_stats_logger = cudf::memory_stats_logger();

state.exec(nvbench::exec_tag::sync | nvbench::exec_tag::timer,
Expand Down
6 changes: 3 additions & 3 deletions cpp/benchmarks/contiguous_split/contiguous_split.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2019-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2019-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -29,7 +29,7 @@ void chunked_pack(cudf::table_view const& src_table, std::vector<cudf::size_type
while (chunked_pack->has_next()) {
(void)chunked_pack->next(user_buffer);
}
stream.synchronize();
stream.sync();
}

template <typename ContigSplitImpl>
Expand Down Expand Up @@ -58,7 +58,7 @@ void contiguous_split_common(nvbench::state& state,
auto const src_table = cudf::table(std::move(src_cols));

auto stream = cudf::get_default_stream();
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
state.add_global_memory_reads<int8_t>(src_table.alloc_size());
state.add_global_memory_writes<int8_t>(src_table.alloc_size());

Expand Down
6 changes: 3 additions & 3 deletions cpp/benchmarks/copying/concatenate.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/
#include <benchmarks/common/generate_input.hpp>
Expand All @@ -26,7 +26,7 @@ static void bench_concatenate(nvbench::state& state)
auto column_views = std::vector<cudf::column_view>(input_columns.begin(), input_columns.end());

auto stream = cudf::get_default_stream();
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
state.add_global_memory_reads<int64_t>(num_rows * num_cols);
state.add_global_memory_writes<int64_t>(num_rows * num_cols);

Expand Down Expand Up @@ -62,7 +62,7 @@ static void bench_concatenate_strings(nvbench::state& state)
auto column_views = std::vector<cudf::column_view>(num_cols, input);

auto stream = cudf::get_default_stream();
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
auto const sv = cudf::strings_column_view(input);
state.add_global_memory_reads<int8_t>(sv.chars_size(stream) * num_cols);
state.add_global_memory_writes<int64_t>(sv.chars_size(stream) * num_cols);
Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/copying/copy_if_else.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -38,7 +38,7 @@ static void bench_copy_if_else(nvbench::state& state, nvbench::type_list<DataTyp
auto const null_bytes = nulls ? 2 * cudf::bitmask_allocation_size_bytes(num_rows) : 0;

auto stream = cudf::get_default_stream();
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
state.add_global_memory_reads<int8_t>(bytes_read);
state.add_global_memory_writes<int8_t>(bytes_written + null_bytes);

Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/copying/gather.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2019-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2019-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -40,7 +40,7 @@ static void bench_gather(nvbench::state& state)
row_count{num_rows});

auto stream = cudf::get_default_stream();
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
state.add_global_memory_reads<int8_t>(source_table->alloc_size());
state.add_global_memory_writes<int8_t>(source_table->alloc_size());

Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/copying/scatter.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2019-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2019-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -41,7 +41,7 @@ static void bench_scatter(nvbench::state& state)
row_count{num_rows});

auto stream = cudf::get_default_stream();
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
state.add_global_memory_reads<int8_t>(source_table->alloc_size());
state.add_global_memory_writes<int8_t>(target_table->alloc_size());

Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/copying/shift.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/
#include <benchmarks/common/generate_input.hpp>
Expand Down Expand Up @@ -35,7 +35,7 @@ static void bench_shift(nvbench::state& state)
auto const null_bytes = use_validity ? 2 * cudf::bitmask_allocation_size_bytes(num_rows) : 0;

auto stream = cudf::get_default_stream();
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
state.add_global_memory_reads<int8_t>(bytes_read);
state.add_global_memory_writes<int8_t>(bytes_written + null_bytes);

Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/decimal/convert_floating.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -117,7 +117,7 @@ void bench_cast_decimal(nvbench::state& state, nvbench::type_list<InputType, Out

// Stream
auto stream = cudf::get_default_stream();
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));

// Run benchmark
auto const mem_stats_logger = cudf::memory_stats_logger();
Expand Down
2 changes: 1 addition & 1 deletion cpp/benchmarks/dictionary/concatenate.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ static void bench_dictionary_concatenate(nvbench::state& state)
auto result = cudf::concatenate(views, stream);
state.add_global_memory_writes<uint8_t>(result->alloc_size());

state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
auto const mem_stats_logger = cudf::memory_stats_logger();

state.exec(nvbench::exec_tag::sync, [&](nvbench::launch&) { cudf::concatenate(views, stream); });
Expand Down
2 changes: 1 addition & 1 deletion cpp/benchmarks/dictionary/encode.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ static void bench_dictionary_encode(nvbench::state& state)
cudf::dictionary::encode(input->view(), cudf::data_type{cudf::type_id::INT32}, stream);
state.add_global_memory_writes<uint8_t>(result->alloc_size());

state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
auto const mem_stats_logger = cudf::memory_stats_logger();

state.exec(nvbench::exec_tag::sync, [&](nvbench::launch&) {
Expand Down
2 changes: 1 addition & 1 deletion cpp/benchmarks/dictionary/match_keys.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ static void bench_dictionary_match_keys(nvbench::state& state)

state.add_global_memory_reads<uint8_t>(input_bytes);

state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
auto const mem_stats_logger = cudf::memory_stats_logger();

state.exec(nvbench::exec_tag::sync, [&](nvbench::launch&) {
Expand Down
2 changes: 1 addition & 1 deletion cpp/benchmarks/dictionary/set_keys.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ static void bench_dictionary_set_keys(nvbench::state& state)
auto result = cudf::dictionary::set_keys(dict_view, new_keys->view(), stream);
state.add_global_memory_writes<uint8_t>(result->alloc_size());

state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
auto const mem_stats_logger = cudf::memory_stats_logger();

state.exec(nvbench::exec_tag::sync, [&](nvbench::launch&) {
Expand Down
2 changes: 1 addition & 1 deletion cpp/benchmarks/dictionary/sort.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ static void bench_dictionary_sort(nvbench::state& state)
auto result = cudf::sort(input_table, {}, {}, stream);
state.add_global_memory_writes<uint8_t>(result->alloc_size());

state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));
auto const mem_stats_logger = cudf::memory_stats_logger();

state.exec(nvbench::exec_tag::sync,
Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/filling/repeat.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -36,7 +36,7 @@ void nvbench_repeat(nvbench::state& state, nvbench::type_list<TypeParam>)

state.add_global_memory_reads(input_table->alloc_size());
state.add_global_memory_writes(output->alloc_size());
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));

auto const mem_stats_logger = cudf::memory_stats_logger();
state.exec(nvbench::exec_tag::sync,
Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/groupby/group_complex_keys.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -114,7 +114,7 @@ void run_benchmark_complex_keys(nvbench::state& state)

auto const mem_stats_logger = cudf::memory_stats_logger();
auto const stream = cudf::get_default_stream();
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(stream.get()));

state.exec(nvbench::exec_tag::sync, [&](nvbench::launch&) {
cudf::groupby::groupby gb_obj(keys_table->view());
Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/groupby/group_histogram.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2022-2025, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -44,7 +44,7 @@ void groupby_histogram_helper(nvbench::state& state,
cudf::make_histogram_aggregation<cudf::groupby_aggregation>());

auto const mem_stats_logger = cudf::memory_stats_logger();
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
state.exec(nvbench::exec_tag::sync, [&](nvbench::launch& launch) {
auto gb_obj = cudf::groupby::groupby(cudf::table_view({keys->view()}));
auto const result = gb_obj.aggregate(requests);
Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/groupby/group_m2_var_std.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -59,7 +59,7 @@ void run_benchmark(nvbench::state& state,
}

auto const mem_stats_logger = cudf::memory_stats_logger();
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
state.exec(nvbench::exec_tag::sync, [&](nvbench::launch&) {
auto gb_obj = cudf::groupby::groupby(cudf::table_view({keys->view()}));
[[maybe_unused]] auto const result = gb_obj.aggregate(requests);
Expand Down
6 changes: 3 additions & 3 deletions cpp/benchmarks/groupby/group_max.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -59,7 +59,7 @@ void groupby_max_helper(nvbench::state& state,
}

auto const mem_stats_logger = cudf::memory_stats_logger();
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
state.exec(nvbench::exec_tag::sync, [&](nvbench::launch& launch) {
auto gb_obj = cudf::groupby::groupby(cudf::table_view({keys_view, keys_view, keys_view}));
auto const result = gb_obj.aggregate(requests);
Expand Down Expand Up @@ -123,7 +123,7 @@ void bench_groupby_max_cardinality(nvbench::state& state, nvbench::type_list<Typ
auto keys_view = keys->view();

auto const mem_stats_logger = cudf::memory_stats_logger();
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));

if (is_streaming) {
std::vector<cudf::column_view> all_columns = {keys_view, keys_view, keys_view};
Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/groupby/group_max_multithreaded.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -71,7 +71,7 @@ void bench_groupby_max_multithreaded(nvbench::state& state, nvbench::type_list<T
threads.detach_sequence(decltype(num_threads){0}, num_threads, perform_agg);
threads.wait();
cudf::detail::join_streams(streams, cudf::get_default_stream());
cudf::get_default_stream().synchronize();
cudf::get_default_stream().sync();
timer.stop();
});

Expand Down
6 changes: 3 additions & 3 deletions cpp/benchmarks/groupby/group_no_requests.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand All @@ -26,7 +26,7 @@ static void bench_groupby_no_requests(nvbench::state& state)

state.add_global_memory_reads<nvbench::int8_t>(keys_table->alloc_size());

state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
auto const mem_stats_logger = cudf::memory_stats_logger();
state.exec(nvbench::exec_tag::sync, [&](nvbench::launch& launch) {
cudf::groupby::groupby gb_obj(*keys_table);
Expand Down Expand Up @@ -56,7 +56,7 @@ static void bench_groupby_pre_sorted_no_requests(nvbench::state& state)
std::vector<cudf::groupby::aggregation_request> requests;
state.add_global_memory_reads<nvbench::int8_t>(keys_table->alloc_size());

state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
auto const mem_stats_logger = cudf::memory_stats_logger();
state.exec(nvbench::exec_tag::sync, [&](nvbench::launch& launch) {
cudf::groupby::groupby gb_obj(*sorted_keys, cudf::null_policy::EXCLUDE, cudf::sorted::YES);
Expand Down
4 changes: 2 additions & 2 deletions cpp/benchmarks/groupby/group_nth.cpp
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION.
* SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*/

Expand Down Expand Up @@ -37,7 +37,7 @@ static void bench_groupby_nth(nvbench::state& state)
state.add_global_memory_reads<nvbench::int8_t>(vals->alloc_size());
std::size_t write_size = 0;

state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
auto const mem_stats_logger = cudf::memory_stats_logger();
state.exec(nvbench::exec_tag::sync, [&](nvbench::launch& launch) {
cudf::groupby::groupby gb_obj(*sorted_keys, cudf::null_policy::EXCLUDE, cudf::sorted::YES);
Expand Down
2 changes: 1 addition & 1 deletion cpp/benchmarks/groupby/group_nunique.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ void bench_groupby_nunique(nvbench::state& state, nvbench::type_list<Type>)
auto const requests = make_aggregation_request_vector(
*vals, cudf::make_nunique_aggregation<cudf::groupby_aggregation>());

state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().value()));
state.set_cuda_stream(nvbench::make_cuda_stream_view(cudf::get_default_stream().get()));
auto const mem_stats_logger = cudf::memory_stats_logger();
state.exec(nvbench::exec_tag::sync, [&](nvbench::launch& launch) {
auto gb_obj =
Expand Down
Loading
Loading