1#include <cassert>
2#include <iostream>
3#include <set>
4#include <vector>
5
6#include "benchmark/benchmark.h"
7
8class ArgsProductFixture : public ::benchmark::Fixture {
9 public:
10  ArgsProductFixture()
11      : expectedValues({{0, 100, 2000, 30000},
12                        {1, 15, 3, 8},
13                        {1, 15, 3, 9},
14                        {1, 15, 7, 8},
15                        {1, 15, 7, 9},
16                        {1, 15, 10, 8},
17                        {1, 15, 10, 9},
18                        {2, 15, 3, 8},
19                        {2, 15, 3, 9},
20                        {2, 15, 7, 8},
21                        {2, 15, 7, 9},
22                        {2, 15, 10, 8},
23                        {2, 15, 10, 9},
24                        {4, 5, 6, 11}}) {}
25
26  void SetUp(const ::benchmark::State& state) override {
27    std::vector<int64_t> ranges = {state.range(0), state.range(1),
28                                   state.range(2), state.range(3)};
29
30    assert(expectedValues.find(ranges) != expectedValues.end());
31
32    actualValues.insert(ranges);
33  }
34
35  // NOTE: This is not TearDown as we want to check after _all_ runs are
36  // complete.
37  ~ArgsProductFixture() override {
38    if (actualValues != expectedValues) {
39      std::cout << "EXPECTED\n";
40      for (const auto& v : expectedValues) {
41        std::cout << "{";
42        for (int64_t iv : v) {
43          std::cout << iv << ", ";
44        }
45        std::cout << "}\n";
46      }
47      std::cout << "ACTUAL\n";
48      for (const auto& v : actualValues) {
49        std::cout << "{";
50        for (int64_t iv : v) {
51          std::cout << iv << ", ";
52        }
53        std::cout << "}\n";
54      }
55    }
56  }
57
58  std::set<std::vector<int64_t>> expectedValues;
59  std::set<std::vector<int64_t>> actualValues;
60};
61
62BENCHMARK_DEFINE_F(ArgsProductFixture, Empty)(benchmark::State& state) {
63  for (auto _ : state) {
64    int64_t product =
65        state.range(0) * state.range(1) * state.range(2) * state.range(3);
66    for (int64_t x = 0; x < product; x++) {
67      benchmark::DoNotOptimize(x);
68    }
69  }
70}
71
72BENCHMARK_REGISTER_F(ArgsProductFixture, Empty)
73    ->Args({0, 100, 2000, 30000})
74    ->ArgsProduct({{1, 2}, {15}, {3, 7, 10}, {8, 9}})
75    ->Args({4, 5, 6, 11});
76
77BENCHMARK_MAIN();
78