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
|
#include <benchmark/benchmark.h>
#include <library/cpp/yson/node/node_builder.h>
using namespace NYT;
static void BM_Map(benchmark::State& state, const std::tuple<int, bool>& input) {
for (auto _ : state) {
TNode node;
TNodeBuilder builder(&node);
if (std::get<1>(input)) {
builder.OnBeginMap(std::get<0>(input));
} else {
builder.OnBeginMap();
}
for (auto i = 0; i < std::get<0>(input); ++i) {
builder.OnKeyedItem(ToString(i));
builder.OnUint64Scalar(i);
}
builder.OnEndMap();
}
}
static void BM_List(benchmark::State& state, const std::tuple<int, bool>& input) {
for (auto _ : state) {
TNode node;
TNodeBuilder builder(&node);
if (std::get<1>(input)) {
builder.OnBeginList(std::get<0>(input));
} else {
builder.OnBeginList();
}
for (auto i = 0; i < std::get<0>(input); ++i) {
builder.OnListItem();
builder.OnUint64Scalar(i);
}
builder.OnEndList();
}
}
BENCHMARK_CAPTURE(BM_Map, map_10, std::make_tuple(10u, false));
BENCHMARK_CAPTURE(BM_Map, map_100, std::make_tuple(100u, false));
BENCHMARK_CAPTURE(BM_Map, map_300, std::make_tuple(300u, false));
BENCHMARK_CAPTURE(BM_Map, map_500, std::make_tuple(500u, false));
BENCHMARK_CAPTURE(BM_Map, map_1000, std::make_tuple(1000u, false));
BENCHMARK_CAPTURE(BM_Map, map_reserve_10, std::make_tuple(10u, true));
BENCHMARK_CAPTURE(BM_Map, map_reserve_100, std::make_tuple(100u, true));
BENCHMARK_CAPTURE(BM_Map, map_reserve_300, std::make_tuple(300u, true));
BENCHMARK_CAPTURE(BM_Map, map_reserve_500, std::make_tuple(500u, true));
BENCHMARK_CAPTURE(BM_Map, map_reserve_1000, std::make_tuple(1000u, true));
BENCHMARK_CAPTURE(BM_List, list_10, std::make_tuple(10u, false));
BENCHMARK_CAPTURE(BM_List, list_100, std::make_tuple(100u, false));
BENCHMARK_CAPTURE(BM_List, list_300, std::make_tuple(300u, false));
BENCHMARK_CAPTURE(BM_List, list_500, std::make_tuple(500u, false));
BENCHMARK_CAPTURE(BM_List, list_1000, std::make_tuple(1000u, false));
BENCHMARK_CAPTURE(BM_List, list_reserve_10, std::make_tuple(10u, true));
BENCHMARK_CAPTURE(BM_List, list_reserve_100, std::make_tuple(100u, true));
BENCHMARK_CAPTURE(BM_List, list_reserve_300, std::make_tuple(300u, true));
BENCHMARK_CAPTURE(BM_List, list_reserve_500, std::make_tuple(500u, true));
BENCHMARK_CAPTURE(BM_List, list_reserve_1000, std::make_tuple(1000u, true));
|