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
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
// SPDX-FileCopyrightText: Copyright Contributors to the Kokkos project
#ifndef KOKKOS_CORE_PERFTEST_BENCHMARK_CONTEXT_HPP
#define KOKKOS_CORE_PERFTEST_BENCHMARK_CONTEXT_HPP
#include <string>
#include <benchmark/benchmark.h>
#include <Kokkos_Macros.hpp>
#ifdef KOKKOS_ENABLE_EXPERIMENTAL_CXX20_MODULES
import kokkos.core;
#else
#include <Kokkos_Core.hpp>
#endif
#include <Kokkos_Version_Info.hpp>
namespace KokkosBenchmark {
/**
* \brief Gather all context information and add it to benchmark context data
*/
void add_benchmark_context(bool verbose = false);
/**
* \brief Mark the label as a figure of merit.
*/
inline std::string benchmark_fom(const std::string& label) {
return "FOM: " + label;
}
/**
* \brief Report throughput and amount of data processed for simple View
* operations
*/
template <class ViewType>
void report_results(benchmark::State& state, ViewType view, int data_ratio,
double time) {
// data processed in megabytes
const double data_processed = static_cast<double>(data_ratio) * view.size() *
sizeof(typename ViewType::value_type) /
1'000'000;
state.SetIterationTime(time);
state.counters["MB"] = benchmark::Counter(data_processed);
state.counters[KokkosBenchmark::benchmark_fom("GB/s")] = benchmark::Counter(
data_processed / 1'000, benchmark::Counter::kIsIterationInvariantRate);
}
} // namespace KokkosBenchmark
#endif
|
