1 #include "benchmark/benchmark_api.h" 2 #include <chrono> 3 #include <thread> 4 5 #if defined(NDEBUG) 6 #undef NDEBUG 7 #endif 8 #include <cassert> 9 10 void BM_basic(benchmark::State& state) { 11 while (state.KeepRunning()) { 12 } 13 } 14 15 void BM_basic_slow(benchmark::State& state) { 16 std::chrono::milliseconds sleep_duration(state.range(0)); 17 while (state.KeepRunning()) { 18 std::this_thread::sleep_for( 19 std::chrono::duration_cast<std::chrono::nanoseconds>(sleep_duration)); 20 } 21 } 22 23 BENCHMARK(BM_basic); 24 BENCHMARK(BM_basic)->Arg(42); 25 BENCHMARK(BM_basic_slow)->Arg(10)->Unit(benchmark::kNanosecond); 26 BENCHMARK(BM_basic_slow)->Arg(100)->Unit(benchmark::kMicrosecond); 27 BENCHMARK(BM_basic_slow)->Arg(1000)->Unit(benchmark::kMillisecond); 28 BENCHMARK(BM_basic)->Range(1, 8); 29 BENCHMARK(BM_basic)->RangeMultiplier(2)->Range(1, 8); 30 BENCHMARK(BM_basic)->DenseRange(10, 15); 31 BENCHMARK(BM_basic)->Args({42, 42}); 32 BENCHMARK(BM_basic)->Ranges({{64, 512}, {64, 512}}); 33 BENCHMARK(BM_basic)->MinTime(0.7); 34 BENCHMARK(BM_basic)->UseRealTime(); 35 BENCHMARK(BM_basic)->ThreadRange(2, 4); 36 BENCHMARK(BM_basic)->ThreadPerCpu(); 37 BENCHMARK(BM_basic)->Repetitions(3); 38 39 void CustomArgs(benchmark::internal::Benchmark* b) { 40 for (int i = 0; i < 10; ++i) { 41 b->Arg(i); 42 } 43 } 44 45 BENCHMARK(BM_basic)->Apply(CustomArgs); 46 47 void BM_explicit_iteration_count(benchmark::State& st) { 48 // Test that benchmarks specified with an explicit iteration count are 49 // only run once. 50 static bool invoked_before = false; 51 assert(!invoked_before); 52 invoked_before = true; 53 54 // Test that the requested iteration count is respected. 55 assert(st.max_iterations == 42); 56 size_t actual_iterations = 0; 57 while (st.KeepRunning()) 58 ++actual_iterations; 59 assert(st.iterations() == st.max_iterations); 60 assert(st.iterations() == 42); 61 62 } 63 BENCHMARK(BM_explicit_iteration_count)->Iterations(42); 64 65 BENCHMARK_MAIN() 66
