static benchmark registration

This commit is contained in:
Carl Pearson
2020-09-04 08:36:14 -06:00
parent 9fda681c37
commit 9fa6f3b811
6 changed files with 60 additions and 46 deletions

View File

@@ -4,24 +4,19 @@
void allreduce(bench::State &state) {
const int rank = bench::world_rank();
const int size = bench::world_size();
const int rank = bench::world_rank();
const int size = bench::world_size();
const size_t sz = 100000;
const size_t sz = 1000;
char *data = new char[sz];
for (auto _ : state) {
MPI_Allreduce(MPI_IN_PLACE, data, sz, MPI_BYTE, MPI_SUM, MPI_COMM_WORLD);
}
char *data = new char[sz];
for (auto _ : state) {
MPI_Allreduce(MPI_IN_PLACE, data, sz, MPI_BYTE, MPI_SUM, MPI_COMM_WORLD);
}
state.set_bytes_processed(sz * size);
delete[] data;
state.set_bytes_processed(sz);
delete[] data;
}
int main(int argc, char **argv) {
bench::init(argc, argv);
bench::register_bench("allreduce", allreduce)->timing_max_rank();
bench::run_benchmarks();
bench::finalize();
}
BENCH(allreduce)->timing_max_rank();
BENCH_MAIN()

View File

@@ -7,10 +7,5 @@ void empty(bench::State &state) {
}
}
int main(int argc, char **argv) {
bench::init(argc, argv);
bench::register_bench("empty", empty)->timing_root_rank()->no_iter_barrier();
bench::run_benchmarks();
bench::finalize();
}
BENCH(empty)->timing_root_rank()->no_iter_barrier();
BENCH_MAIN()

View File

@@ -27,9 +27,5 @@ void pingpong(bench::State &state) {
delete[] rbuf;
}
int main(int argc, char **argv) {
bench::init(argc, argv);
bench::register_bench("pingpong", pingpong)->timing_root_rank()->no_iter_barrier();
bench::run_benchmarks();
bench::finalize();
}
BENCH(pingpong)->timing_root_rank()->no_iter_barrier();
BENCH_MAIN()