Add bytes_per_second to copy_if_else benchmark

Adds `bytes_per_second to `COPY_IF_ELSE_BENCH`. This patch relates to rapidsai#13735.
Blonck · Aug 25, 2023 · 808fdc1 · 808fdc1
1 parent 6095a92
commit 808fdc1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/cpp/benchmarks/copying/copy_if_else.cpp b/cpp/benchmarks/copying/copy_if_else.cpp
@@ -47,6 +47,14 @@ static void BM_copy_if_else(benchmark::State& state, bool nulls)
     cuda_event_timer raii(state, true, cudf::get_default_stream());
     cudf::copy_if_else(lhs, rhs, decision);
   }
+
+  auto const bytes_read    = n_rows * (sizeof(TypeParam) + sizeof(bool));
+  auto const bytes_written = n_rows * sizeof(TypeParam);
+  auto const null_bytes    = nulls ? 2 * cudf::bitmask_allocation_size_bytes(n_rows) : 0;
+
+  // Use number of bytes read and written.
+  state.SetBytesProcessed(static_cast<int64_t>(state.iterations()) *
+                          (bytes_read + bytes_written + null_bytes));
 }
 
 #define COPY_BENCHMARK_DEFINE(name, type, b)                  \