| #include <benchmark/benchmark.h> |
| |
| #include <unistd.h> |
| |
| #include <pthreadpool.h> |
| |
| |
| static void SetNumberOfThreads(benchmark::internal::Benchmark* benchmark) { |
| const int max_threads = sysconf(_SC_NPROCESSORS_ONLN); |
| for (int t = 1; t <= max_threads; t++) { |
| benchmark->Arg(t); |
| } |
| } |
| |
| |
| static void compute_1d(void*, size_t x) { |
| } |
| |
| static void pthreadpool_parallelize_1d(benchmark::State& state) { |
| const uint32_t threads = static_cast<uint32_t>(state.range(0)); |
| pthreadpool_t threadpool = pthreadpool_create(threads); |
| while (state.KeepRunning()) { |
| pthreadpool_parallelize_1d( |
| threadpool, |
| compute_1d, |
| nullptr /* context */, |
| threads, |
| 0 /* flags */); |
| } |
| pthreadpool_destroy(threadpool); |
| } |
| BENCHMARK(pthreadpool_parallelize_1d)->UseRealTime()->Apply(SetNumberOfThreads); |
| |
| |
| static void compute_1d_tile_1d(void*, size_t, size_t) { |
| } |
| |
| static void pthreadpool_parallelize_1d_tile_1d(benchmark::State& state) { |
| const uint32_t threads = static_cast<uint32_t>(state.range(0)); |
| pthreadpool_t threadpool = pthreadpool_create(threads); |
| while (state.KeepRunning()) { |
| pthreadpool_parallelize_1d_tile_1d( |
| threadpool, |
| compute_1d_tile_1d, |
| nullptr /* context */, |
| threads, 1, |
| 0 /* flags */); |
| } |
| pthreadpool_destroy(threadpool); |
| } |
| BENCHMARK(pthreadpool_parallelize_1d_tile_1d)->UseRealTime()->Apply(SetNumberOfThreads); |
| |
| |
| static void compute_2d(void*, size_t, size_t) { |
| } |
| |
| static void pthreadpool_parallelize_2d(benchmark::State& state) { |
| const uint32_t threads = static_cast<uint32_t>(state.range(0)); |
| pthreadpool_t threadpool = pthreadpool_create(threads); |
| while (state.KeepRunning()) { |
| pthreadpool_parallelize_2d( |
| threadpool, |
| compute_2d, |
| nullptr /* context */, |
| 1, threads, |
| 0 /* flags */); |
| } |
| pthreadpool_destroy(threadpool); |
| } |
| BENCHMARK(pthreadpool_parallelize_2d)->UseRealTime()->Apply(SetNumberOfThreads); |
| |
| |
| static void compute_2d_tile_2d(void*, size_t, size_t, size_t, size_t) { |
| } |
| |
| static void pthreadpool_parallelize_2d_tile_2d(benchmark::State& state) { |
| const uint32_t threads = static_cast<uint32_t>(state.range(0)); |
| pthreadpool_t threadpool = pthreadpool_create(threads); |
| while (state.KeepRunning()) { |
| pthreadpool_parallelize_2d_tile_2d( |
| threadpool, |
| compute_2d_tile_2d, |
| nullptr /* context */, |
| 1, threads, |
| 1, 1, |
| 0 /* flags */); |
| } |
| pthreadpool_destroy(threadpool); |
| } |
| BENCHMARK(pthreadpool_parallelize_2d_tile_2d)->UseRealTime()->Apply(SetNumberOfThreads); |
| |
| |
| BENCHMARK_MAIN(); |