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
|
#include "purify/config.h"
#include <benchmark/benchmark.h>
#include <sopt/mpi/communicator.h>
#include <sopt/mpi/session.h>
// This reporter does nothing.
// We can use it to disable output from all but the root process
class NullReporter : public ::benchmark::BenchmarkReporter {
public:
NullReporter() {}
virtual bool ReportContext(const Context &) { return true; }
virtual void ReportRuns(const std::vector<Run> &) {}
virtual void Finalize() {}
};
// The main is rewritten to allow for MPI initializing and for selecting a
// reporter according to the process rank
int main(int argc, char const **argv) {
#ifdef PURIFY_MPI
auto const session = sopt::mpi::init(argc, argv);
auto const world = sopt::mpi::Communicator::World();
#endif
::benchmark::Initialize(&argc, const_cast<char **>(argv));
#ifdef PURIFY_MPI
if (world.is_root())
// root process will use a reporter from the usual set provided by
// ::benchmark
::benchmark::RunSpecifiedBenchmarks();
else {
// reporting from other processes is disabled by passing a custom reporter
NullReporter null;
::benchmark::RunSpecifiedBenchmarks(&null);
}
#endif
return 0;
}
|
