1 #include <cassert>
2 #include <iostream>
3 #include <set>
4 #include <vector>
5
6 #include "benchmark/benchmark.h"
7
8 class MultipleRangesFixture : public ::benchmark::Fixture {
9 public:
MultipleRangesFixture()10 MultipleRangesFixture()
11 : expectedValues({{1, 3, 5},
12 {1, 3, 8},
13 {1, 3, 15},
14 {2, 3, 5},
15 {2, 3, 8},
16 {2, 3, 15},
17 {1, 4, 5},
18 {1, 4, 8},
19 {1, 4, 15},
20 {2, 4, 5},
21 {2, 4, 8},
22 {2, 4, 15},
23 {1, 7, 5},
24 {1, 7, 8},
25 {1, 7, 15},
26 {2, 7, 5},
27 {2, 7, 8},
28 {2, 7, 15},
29 {7, 6, 3}}) {}
30
SetUp(const::benchmark::State & state)31 void SetUp(const ::benchmark::State& state) BENCHMARK_OVERRIDE {
32 std::vector<int64_t> ranges = {state.range(0), state.range(1),
33 state.range(2)};
34
35 assert(expectedValues.find(ranges) != expectedValues.end());
36
37 actualValues.insert(ranges);
38 }
39
40 // NOTE: This is not TearDown as we want to check after _all_ runs are
41 // complete.
~MultipleRangesFixture()42 virtual ~MultipleRangesFixture() {
43 if (actualValues != expectedValues) {
44 std::cout << "EXPECTED\n";
45 for (const auto& v : expectedValues) {
46 std::cout << "{";
47 for (int64_t iv : v) {
48 std::cout << iv << ", ";
49 }
50 std::cout << "}\n";
51 }
52 std::cout << "ACTUAL\n";
53 for (const auto& v : actualValues) {
54 std::cout << "{";
55 for (int64_t iv : v) {
56 std::cout << iv << ", ";
57 }
58 std::cout << "}\n";
59 }
60 }
61 }
62
63 std::set<std::vector<int64_t>> expectedValues;
64 std::set<std::vector<int64_t>> actualValues;
65 };
66
BENCHMARK_DEFINE_F(MultipleRangesFixture,Empty)67 BENCHMARK_DEFINE_F(MultipleRangesFixture, Empty)(benchmark::State& state) {
68 for (auto _ : state) {
69 int64_t product = state.range(0) * state.range(1) * state.range(2);
70 for (int64_t x = 0; x < product; x++) {
71 benchmark::DoNotOptimize(x);
72 }
73 }
74 }
75
76 BENCHMARK_REGISTER_F(MultipleRangesFixture, Empty)
77 ->RangeMultiplier(2)
78 ->Ranges({{1, 2}, {3, 7}, {5, 15}})
79 ->Args({7, 6, 3});
80
BM_CheckDefaultArgument(benchmark::State & state)81 void BM_CheckDefaultArgument(benchmark::State& state) {
82 // Test that the 'range()' without an argument is the same as 'range(0)'.
83 assert(state.range() == state.range(0));
84 assert(state.range() != state.range(1));
85 for (auto _ : state) {
86 }
87 }
88 BENCHMARK(BM_CheckDefaultArgument)->Ranges({{1, 5}, {6, 10}});
89
BM_MultipleRanges(benchmark::State & st)90 static void BM_MultipleRanges(benchmark::State& st) {
91 for (auto _ : st) {
92 }
93 }
94 BENCHMARK(BM_MultipleRanges)->Ranges({{5, 5}, {6, 6}});
95
96 BENCHMARK_MAIN();
97