Function BM_NAME

tensorflow/compiler/aot/test.cc:73–88

Simple benchmark that repeatedly runs the generated function.

// Simple benchmark that repeatedly runs the generated function.
void BM_NAME(int iters) {
  testing::StopTiming();

  Eigen::ThreadPool pool(port::MaxParallelism());
  Eigen::ThreadPoolDevice device(&pool, pool.NumThreads());

  CPP_CLASS computation;
  computation.set_thread_pool(&device);
  zero_buffers(&computation);

  testing::StartTiming();
  while (--iters) {
    computation.Run();
  }
  testing::StopTiming();
}
BENCHMARK(BM_NAME);

}  // namespace