From d5ba8420bd00e390209c49dc91ef08be9f31840e Mon Sep 17 00:00:00 2001
From: AJ Heller <hork@google.com>
Date: Wed, 16 Oct 2024 18:34:06 +0000
Subject: [PATCH] [benchmark] Measure all threads CPU in callback benchmarks

---
 test/cpp/microbenchmarks/bm_callback_streaming_ping_pong.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test/cpp/microbenchmarks/bm_callback_streaming_ping_pong.cc b/test/cpp/microbenchmarks/bm_callback_streaming_ping_pong.cc
index 2188ae49333..e7f7fef1964 100644
--- a/test/cpp/microbenchmarks/bm_callback_streaming_ping_pong.cc
+++ b/test/cpp/microbenchmarks/bm_callback_streaming_ping_pong.cc
@@ -47,6 +47,7 @@ static void StreamingPingPongArgs(benchmark::internal::Benchmark* b) {
        msg_size == 0 ? msg_size++ : msg_size *= 8) {
     b->Args({msg_size, 1});
     b->Args({msg_size, 2});
+    b->MeasureProcessCPUTime()->UseRealTime();
   }
 }