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
|
// (C) Copyright Victor Ananyev 2021.
// Use, modification and distribution are subject to the
// Boost Software License, Version 1.0. (See accompanying file
// LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
#include <random>
#include <vector>
#include <boost/math/distributions.hpp>
#include <benchmark/benchmark.h>
template <class Z, template<typename> class dist >
void test_mode_2param(benchmark::State& state)
{
using boost::math::normal_distribution;
std::random_device rd;
std::mt19937_64 mt(rd());
std::normal_distribution<Z> noise(0., 1E-6);
for (auto _ : state)
{
state.PauseTiming();
Z p1 = state.range(0) + noise(mt);
Z p2 = state.range(1) + noise(mt);
dist<Z> the_dist(p1, p2);
state.ResumeTiming();
try {
benchmark::DoNotOptimize(mode(the_dist));
}
catch (boost::wrapexcept<boost::math::evaluation_error>& e) {
state.SkipWithError(e.what());
break;
}
}
}
static void fixed_ratio_2args(benchmark::internal::Benchmark* b, long double left_div_right, std::vector<int64_t> lefts) {
for (const long double &left: lefts) {
b->Args({static_cast<int64_t>(left), static_cast<int64_t>((left/left_div_right))});
}
}
using boost::math::non_central_chi_squared_distribution;
BENCHMARK_TEMPLATE(test_mode_2param, long double, non_central_chi_squared_distribution)->ArgsProduct({
{2, 15, 50},
benchmark::CreateRange(4, 1024, /*multi=*/2)
})->Name("fixed_k");
BENCHMARK_TEMPLATE(test_mode_2param, long double, non_central_chi_squared_distribution)->ArgsProduct({
benchmark::CreateRange(4, 4096, /*multi=*/2),
{1, 30, 100, 500}
})->Name("fixed_nc");
BENCHMARK_TEMPLATE(test_mode_2param, long double, non_central_chi_squared_distribution)
-> Apply([](benchmark::internal::Benchmark*b) {
fixed_ratio_2args(b, 0.05, benchmark::CreateRange(4, 4096, /*multi=*/2));
}) -> Name("fixed_scale_0_05");
BENCHMARK_TEMPLATE(test_mode_2param, long double, non_central_chi_squared_distribution)
-> Apply([](benchmark::internal::Benchmark*b) {
fixed_ratio_2args(b, 0.15, benchmark::CreateRange(4, 4096, /*multi=*/2));
}) -> Name("fixed_scale_0_15");
BENCHMARK_TEMPLATE(test_mode_2param, long double, non_central_chi_squared_distribution)
-> Apply([](benchmark::internal::Benchmark*b) {
fixed_ratio_2args(b, 0.25, benchmark::CreateRange(4, 4096, /*multi=*/2));
}) -> Name("fixed_scale_0_25");
BENCHMARK_MAIN();
|
