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 | |
21 | BENCHMARK_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 | } |
30 | BENCHMARK_REGISTER_F(Schedule, Empty)->Apply(Schedule::args); |
31 | |
32 | BENCHMARK_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 | } |
48 | BENCHMARK_REGISTER_F(Schedule, SomeWork)->Apply(Schedule::args); |
49 | |
50 | BENCHMARK_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 | } |
69 | BENCHMARK_REGISTER_F(Schedule, SomeWorkWorkerAffinityOneOf) |
70 | ->Apply(Schedule::args); |
71 | |