1// Copyright 2020 The Marl Authors.
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// https://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15#include "marl_bench.h"
16
17#include "marl/waitgroup.h"
18
19#include "benchmark/benchmark.h"
20
21BENCHMARK_DEFINE_F(Schedule, Empty)(benchmark::State& state) {
22 run(state, [&](int numTasks) {
23 for (auto _ : state) {
24 for (auto i = 0; i < numTasks; i++) {
25 marl::schedule([] {});
26 }
27 }
28 });
29}
30BENCHMARK_REGISTER_F(Schedule, Empty)->Apply(Schedule::args);
31
32BENCHMARK_DEFINE_F(Schedule, SomeWork)
33(benchmark::State& state) {
34 run(state, [&](int numTasks) {
35 for (auto _ : state) {
36 marl::WaitGroup wg;
37 wg.add(numTasks);
38 for (auto i = 0; i < numTasks; i++) {
39 marl::schedule([=] {
40 benchmark::DoNotOptimize(doSomeWork(i));
41 wg.done();
42 });
43 }
44 wg.wait();
45 }
46 });
47}
48BENCHMARK_REGISTER_F(Schedule, SomeWork)->Apply(Schedule::args);
49
50BENCHMARK_DEFINE_F(Schedule, SomeWorkWorkerAffinityOneOf)
51(benchmark::State& state) {
52 marl::Scheduler::Config cfg;
53 cfg.setWorkerThreadAffinityPolicy(
54 marl::Thread::Affinity::Policy::oneOf(marl::Thread::Affinity::all()));
55 run(state, cfg, [&](int numTasks) {
56 for (auto _ : state) {
57 marl::WaitGroup wg;
58 wg.add(numTasks);
59 for (auto i = 0; i < numTasks; i++) {
60 marl::schedule([=] {
61 benchmark::DoNotOptimize(doSomeWork(i));
62 wg.done();
63 });
64 }
65 wg.wait();
66 }
67 });
68}
69BENCHMARK_REGISTER_F(Schedule, SomeWorkWorkerAffinityOneOf)
70 ->Apply(Schedule::args);
71