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
|
/***************************************************************************
* Copyright (c) 2016, Johan Mabille, Sylvain Corlay and Wolf Vollprecht *
* *
* Distributed under the terms of the BSD 3-Clause License. *
* *
* The full license is in the file LICENSE, distributed with this software. *
****************************************************************************/
#ifndef BENCHMARK_SHAPE_HPP
#define BENCHMARK_SHAPE_HPP
#include <benchmark/benchmark.h>
#include "xtensor/xshape.hpp"
#include "xtensor/xstorage.hpp"
namespace xt
{
namespace benchmark_xshape
{
template <class T>
void xshape_initializer(benchmark::State& state)
{
for (auto _ : state)
{
T sv({2, 3, 1});
benchmark::DoNotOptimize(sv.data());
}
}
template <class T>
void xshape_initializer_long(benchmark::State& state)
{
for (auto _ : state)
{
T sv({2, 3, 1, 2, 6, 1, 2, 3, 45, 6, 12, 3, 5, 45, 5, 6});
benchmark::DoNotOptimize(sv.data());
}
}
template <class T>
void xshape_access(benchmark::State& state)
{
T a({3, 2, 1, 3});
for (auto _ : state)
{
a[0] = a[1] * a[2] + a[3];
a[3] = a[1];
a[1] = a[2] + a[3];
a[2] = a[3];
benchmark::DoNotOptimize(a.data());
}
}
BENCHMARK_TEMPLATE(xshape_initializer, std::vector<std::size_t>);
BENCHMARK_TEMPLATE(xshape_initializer, xt::svector<std::size_t, 4>);
BENCHMARK_TEMPLATE(xshape_initializer, std::array<std::size_t, 4>);
BENCHMARK_TEMPLATE(xshape_initializer_long, xt::svector<std::size_t, 4>);
BENCHMARK_TEMPLATE(xshape_initializer_long, xt::uvector<std::size_t>);
BENCHMARK_TEMPLATE(xshape_initializer_long, std::vector<std::size_t>);
BENCHMARK_TEMPLATE(xshape_access, xt::uvector<double>);
BENCHMARK_TEMPLATE(xshape_access, std::vector<double>);
BENCHMARK_TEMPLATE(xshape_access, xt::svector<std::size_t, 4>);
BENCHMARK_TEMPLATE(xshape_access, std::array<std::size_t, 4>);
}
}
#endif
|
