Add find / count benchmark (#4387)

Co-authored-by: Stephan T. Lavavej <[email protected]>

benchmarks/CMakeLists.txt

@@ -109,6 +109,7 @@ function(add_benchmark name)
 endfunction()
 
 add_benchmark(bitset_to_string src/bitset_to_string.cpp)
+add_benchmark(find_and_count src/find_and_count.cpp)
 add_benchmark(locale_classic src/locale_classic.cpp)
 add_benchmark(minmax_element src/minmax_element.cpp)
 add_benchmark(path_lexically_normal src/path_lexically_normal.cpp)

benchmarks/src/find_and_count.cpp

@@ -0,0 +1,72 @@
+// Copyright (c) Microsoft Corporation.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+#include <algorithm>
+#include <benchmark/benchmark.h>
+#include <cstddef>
+#include <cstdint>
+#include <ranges>
+
+enum class Op {
+    FindSized,
+    FindUnsized,
+    Count,
+};
+
+using namespace std;
+
+template <class T, size_t Size, size_t Pos, Op Operation>
+void bm(benchmark::State& state) {
+    T a[Size];
+
+    fill_n(a, Size, T{'0'});
+    if constexpr (Pos < Size) {
+        a[Pos] = T{'1'};
+    } else {
+        static_assert(Operation != Op::FindUnsized);
+    }
+
+    for (auto _ : state) {
+        if constexpr (Operation == Op::FindSized) {
+            benchmark::DoNotOptimize(ranges::find(a, a + Size, T{'1'}));
+        } else if constexpr (Operation == Op::FindUnsized) {
+            benchmark::DoNotOptimize(ranges::find(a, unreachable_sentinel, T{'1'}));
+        } else if constexpr (Operation == Op::Count) {
+            benchmark::DoNotOptimize(ranges::count(a, a + Size, T{'1'}));
+        }
+    }
+}
+
+BENCHMARK(bm<uint8_t, 8021, 3056, Op::FindSized>);
+BENCHMARK(bm<uint8_t, 8021, 3056, Op::FindUnsized>);
+BENCHMARK(bm<uint8_t, 8021, 3056, Op::Count>);
+
+BENCHMARK(bm<uint16_t, 8021, 3056, Op::FindSized>);
+BENCHMARK(bm<uint16_t, 8021, 3056, Op::FindUnsized>);
+BENCHMARK(bm<uint16_t, 8021, 3056, Op::Count>);
+
+BENCHMARK(bm<uint32_t, 8021, 3056, Op::FindSized>);
+BENCHMARK(bm<uint32_t, 8021, 3056, Op::FindUnsized>);
+BENCHMARK(bm<uint32_t, 8021, 3056, Op::Count>);
+
+BENCHMARK(bm<uint64_t, 8021, 3056, Op::FindSized>);
+BENCHMARK(bm<uint64_t, 8021, 3056, Op::FindUnsized>);
+BENCHMARK(bm<uint64_t, 8021, 3056, Op::Count>);
+
+BENCHMARK(bm<int8_t, 8021, 3056, Op::FindSized>);
+BENCHMARK(bm<int8_t, 8021, 3056, Op::FindUnsized>);
+BENCHMARK(bm<int8_t, 8021, 3056, Op::Count>);
+
+BENCHMARK(bm<int16_t, 8021, 3056, Op::FindSized>);
+BENCHMARK(bm<int16_t, 8021, 3056, Op::FindUnsized>);
+BENCHMARK(bm<int16_t, 8021, 3056, Op::Count>);
+
+BENCHMARK(bm<int32_t, 8021, 3056, Op::FindSized>);
+BENCHMARK(bm<int32_t, 8021, 3056, Op::FindUnsized>);
+BENCHMARK(bm<int32_t, 8021, 3056, Op::Count>);
+
+BENCHMARK(bm<int64_t, 8021, 3056, Op::FindSized>);
+BENCHMARK(bm<int64_t, 8021, 3056, Op::FindUnsized>);
+BENCHMARK(bm<int64_t, 8021, 3056, Op::Count>);
+
+BENCHMARK_MAIN();