Eric Fiselier | d9b9ef7 | 2016-07-19 23:07:03 +0000 | [diff] [blame] | 1 | #include <unordered_set> |
| 2 | #include <vector> |
| 3 | #include <cstdint> |
| 4 | |
| 5 | #include "benchmark/benchmark_api.h" |
| 6 | #include "GenerateInput.hpp" |
| 7 | |
| 8 | constexpr std::size_t TestNumInputs = 1024; |
| 9 | |
| 10 | template <class GenInputs> |
| 11 | void BM_Sort(benchmark::State& st, GenInputs gen) { |
| 12 | using ValueType = typename decltype(gen(0))::value_type; |
Eric Fiselier | 30b48cb | 2016-08-09 18:56:48 +0000 | [diff] [blame] | 13 | const auto in = gen(st.range(0)); |
Eric Fiselier | d9b9ef7 | 2016-07-19 23:07:03 +0000 | [diff] [blame] | 14 | std::vector<ValueType> inputs[5]; |
| 15 | auto reset_inputs = [&]() { |
| 16 | for (auto& C : inputs) { |
| 17 | C = in; |
| 18 | benchmark::DoNotOptimize(C.data()); |
| 19 | } |
| 20 | }; |
| 21 | reset_inputs(); |
| 22 | while (st.KeepRunning()) { |
| 23 | for (auto& I : inputs) { |
| 24 | std::sort(I.data(), I.data() + I.size()); |
| 25 | benchmark::DoNotOptimize(I.data()); |
| 26 | } |
| 27 | st.PauseTiming(); |
| 28 | reset_inputs(); |
| 29 | benchmark::ClobberMemory(); |
| 30 | st.ResumeTiming(); |
| 31 | } |
| 32 | } |
| 33 | |
| 34 | BENCHMARK_CAPTURE(BM_Sort, random_uint32, |
| 35 | getRandomIntegerInputs<uint32_t>)->Arg(TestNumInputs); |
| 36 | |
| 37 | BENCHMARK_CAPTURE(BM_Sort, sorted_ascending_uint32, |
| 38 | getSortedIntegerInputs<uint32_t>)->Arg(TestNumInputs); |
| 39 | |
| 40 | BENCHMARK_CAPTURE(BM_Sort, sorted_descending_uint32, |
| 41 | getReverseSortedIntegerInputs<uint32_t>)->Arg(TestNumInputs); |
| 42 | |
| 43 | BENCHMARK_CAPTURE(BM_Sort, single_element_uint32, |
| 44 | getDuplicateIntegerInputs<uint32_t>)->Arg(TestNumInputs); |
| 45 | |
| 46 | BENCHMARK_CAPTURE(BM_Sort, pipe_organ_uint32, |
| 47 | getPipeOrganIntegerInputs<uint32_t>)->Arg(TestNumInputs); |
| 48 | |
| 49 | BENCHMARK_CAPTURE(BM_Sort, random_strings, |
| 50 | getRandomStringInputs)->Arg(TestNumInputs); |
| 51 | |
| 52 | BENCHMARK_CAPTURE(BM_Sort, sorted_ascending_strings, |
| 53 | getSortedStringInputs)->Arg(TestNumInputs); |
| 54 | |
| 55 | BENCHMARK_CAPTURE(BM_Sort, sorted_descending_strings, |
| 56 | getReverseSortedStringInputs)->Arg(TestNumInputs); |
| 57 | |
| 58 | BENCHMARK_CAPTURE(BM_Sort, single_element_strings, |
| 59 | getDuplicateStringInputs)->Arg(TestNumInputs); |
| 60 | |
| 61 | |
| 62 | BENCHMARK_MAIN() |