78 lines · plain
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 are36 // 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