File: lexicographical_compare_three_way.bench.cpp

package info (click to toggle)
llvm-toolchain-19 1%3A19.1.7-3~deb12u1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm-proposed-updates
  • size: 1,998,492 kB
  • sloc: cpp: 6,951,680; ansic: 1,486,157; asm: 913,598; python: 232,024; f90: 80,126; objc: 75,281; lisp: 37,276; pascal: 16,990; sh: 10,009; ml: 5,058; perl: 4,724; awk: 3,523; makefile: 3,167; javascript: 2,504; xml: 892; fortran: 664; cs: 573
file content (96 lines) | stat: -rw-r--r-- 3,486 bytes parent folder | download | duplicates (8)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
//===----------------------------------------------------------------------===//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include <algorithm>

#include "benchmark/benchmark.h"
#include "test_iterators.h"

static void BM_lexicographical_compare_three_way_slow_path(benchmark::State& state) {
  auto size = state.range(0);
  std::vector<int> v1;
  v1.resize(size);
  // v2 is identical except for the last value.
  // This means, that `lexicographical_compare_three_way` actually has to
  // compare the complete vector and cannot bail out early.
  std::vector<int> v2 = v1;
  v2.back() += 1;
  int* b1 = v1.data();
  int* e1 = b1 + v1.size();
  int* b2 = v2.data();
  int* e2 = b2 + v2.size();

  for (auto _ : state) {
    auto cmp = std::compare_three_way();
    benchmark::DoNotOptimize(std::__lexicographical_compare_three_way_slow_path(b1, e1, b2, e2, cmp));
  }
}

BENCHMARK(BM_lexicographical_compare_three_way_slow_path)->RangeMultiplier(4)->Range(1, 1 << 20);

static void BM_lexicographical_compare_three_way_fast_path(benchmark::State& state) {
  auto size = state.range(0);
  std::vector<int> v1;
  v1.resize(size);
  // v2 is identical except for the last value.
  // This means, that `lexicographical_compare_three_way` actually has to
  // compare the complete vector and cannot bail out early.
  std::vector<int> v2 = v1;
  v2.back() += 1;
  int* b1 = v1.data();
  int* e1 = b1 + v1.size();
  int* b2 = v2.data();
  int* e2 = b2 + v2.size();

  for (auto _ : state) {
    auto cmp = std::compare_three_way();
    benchmark::DoNotOptimize(std::__lexicographical_compare_three_way_fast_path(b1, e1, b2, e2, cmp));
  }
}

BENCHMARK(BM_lexicographical_compare_three_way_fast_path)->RangeMultiplier(4)->Range(1, 1 << 20);

template <class IteratorT>
static void BM_lexicographical_compare_three_way(benchmark::State& state) {
  auto size = state.range(0);
  std::vector<int> v1;
  v1.resize(size);
  // v2 is identical except for the last value.
  // This means, that `lexicographical_compare_three_way` actually has to
  // compare the complete vector and cannot bail out early.
  std::vector<int> v2 = v1;
  v2.back() += 1;
  auto b1 = IteratorT{v1.data()};
  auto e1 = IteratorT{v1.data() + v1.size()};
  auto b2 = IteratorT{v2.data()};
  auto e2 = IteratorT{v2.data() + v2.size()};

  for (auto _ : state) {
    benchmark::DoNotOptimize(std::lexicographical_compare_three_way(b1, e1, b2, e2));
  }
}

// Type alias to make sure the `*` does not appear in the benchmark name.
// A `*` would confuse the Python test runner running this google benchmark.
using IntPtr = int*;

// `lexicographical_compare_three_way` has a fast path for random access iterators.
BENCHMARK_TEMPLATE(BM_lexicographical_compare_three_way, IntPtr)->RangeMultiplier(4)->Range(1, 1 << 20);
BENCHMARK_TEMPLATE(BM_lexicographical_compare_three_way, random_access_iterator<IntPtr>)
    ->RangeMultiplier(4)
    ->Range(1, 1 << 20);
BENCHMARK_TEMPLATE(BM_lexicographical_compare_three_way, cpp17_input_iterator<IntPtr>)
    ->RangeMultiplier(4)
    ->Range(1, 1 << 20);

int main(int argc, char** argv) {
  benchmark::Initialize(&argc, argv);
  if (benchmark::ReportUnrecognizedArguments(argc, argv))
    return 1;

  benchmark::RunSpecifiedBenchmarks();
}