Function BM_BlockingCounter

tensorflow/core/lib/core/blocking_counter_test.cc:52–71

}  // namespace

static void BM_BlockingCounter(int iters, int num_threads,
                               int shards_per_thread) {
  testing::StopTiming();
  std::unique_ptr<thread::ThreadPool> thread_pool(
      new thread::ThreadPool(Env::Default(), "test", num_threads));
  const int num_shards = num_threads * shards_per_thread;
  testing::StartTiming();
  for (int i = 0; i < iters; ++i) {
    BlockingCounter bc(num_shards);
    for (int j = 0; j < num_threads; ++j) {
      thread_pool->Schedule([&bc, shards_per_thread] {
        for (int k = 0; k < shards_per_thread; ++k) {
          bc.DecrementCount();
        }
      });
    }
    bc.Wait();
  }
  testing::StopTiming();
}

BENCHMARK(BM_BlockingCounter)->RangePair(1, 12, 1, 1000);