1#include "benchmark/benchmark.h"
2#include <chrono>
3#include <thread>
4
5#if defined(NDEBUG)
6#undef NDEBUG
7#endif
8#include <cassert>
9
10void BM_basic(benchmark::State& state) {
11  for (auto _ : state) {
12  }
13}
14
15void BM_basic_slow(benchmark::State& state) {
16  std::chrono::milliseconds sleep_duration(state.range(0));
17  for (auto _ : state) {
18    std::this_thread::sleep_for(
19        std::chrono::duration_cast<std::chrono::nanoseconds>(sleep_duration));
20  }
21}
22
23BENCHMARK(BM_basic);
24BENCHMARK(BM_basic)->Arg(42);
25BENCHMARK(BM_basic_slow)->Arg(10)->Unit(benchmark::kNanosecond);
26BENCHMARK(BM_basic_slow)->Arg(100)->Unit(benchmark::kMicrosecond);
27BENCHMARK(BM_basic_slow)->Arg(1000)->Unit(benchmark::kMillisecond);
28BENCHMARK(BM_basic)->Range(1, 8);
29BENCHMARK(BM_basic)->RangeMultiplier(2)->Range(1, 8);
30BENCHMARK(BM_basic)->DenseRange(10, 15);
31BENCHMARK(BM_basic)->Args({42, 42});
32BENCHMARK(BM_basic)->Ranges({{64, 512}, {64, 512}});
33BENCHMARK(BM_basic)->MinTime(0.7);
34BENCHMARK(BM_basic)->UseRealTime();
35BENCHMARK(BM_basic)->ThreadRange(2, 4);
36BENCHMARK(BM_basic)->ThreadPerCpu();
37BENCHMARK(BM_basic)->Repetitions(3);
38
39void CustomArgs(benchmark::internal::Benchmark* b) {
40  for (int i = 0; i < 10; ++i) {
41    b->Arg(i);
42  }
43}
44
45BENCHMARK(BM_basic)->Apply(CustomArgs);
46
47void BM_explicit_iteration_count(benchmark::State& state) {
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(state.max_iterations == 42);
56  size_t actual_iterations = 0;
57  for (auto _ : state)
58    ++actual_iterations;
59  assert(state.iterations() == state.max_iterations);
60  assert(state.iterations() == 42);
61
62}
63BENCHMARK(BM_explicit_iteration_count)->Iterations(42);
64
65BENCHMARK_MAIN();
66