Add bytes_per_second to hash_partition benchmark

Adds `bytes_per_second` to the `PARTITION_BENCH` benchmark. This patch relates to rapidsai#13735.
Blonck · Aug 25, 2023 · 17fe3c7 · 17fe3c7
1 parent aef903c
commit 17fe3c7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/cpp/benchmarks/hashing/partition.cpp b/cpp/benchmarks/hashing/partition.cpp
@@ -43,6 +43,13 @@ void BM_hash_partition(benchmark::State& state)
     cuda_event_timer timer(state, true);
     auto output = cudf::hash_partition(input, columns_to_hash, num_partitions);
   }
+
+  auto const bytes_read      = num_rows * num_cols * sizeof(T);
+  auto const bytes_written   = num_rows * num_cols * sizeof(T);
+  auto const partition_bytes = num_partitions * sizeof(cudf::size_type);
+
+  state.SetBytesProcessed(static_cast<int64_t>(state.iterations()) *
+                          (bytes_read + bytes_written + partition_bytes));
 }
 
 BENCHMARK_DEFINE_F(Hashing, hash_partition)