forked from taskflow/taskflow
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtaskflow.cpp
More file actions
42 lines (32 loc) · 1.04 KB
/
taskflow.cpp
File metadata and controls
42 lines (32 loc) · 1.04 KB
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
#include "binary_tree.hpp"
#include <taskflow/taskflow.hpp>
// binary_tree_taskflow
void binary_tree_taskflow(size_t num_layers, unsigned num_threads) {
std::atomic<size_t> counter {0};
std::vector<tf::Task> tasks(1 << num_layers);
tf::Executor executor(num_threads);
tf::Taskflow taskflow;
for(unsigned i=1; i<tasks.size(); i++) {
tasks[i] = taskflow.emplace([&](){
counter.fetch_add(1, std::memory_order_relaxed);
});
}
for(unsigned i=1; i<tasks.size(); i++) {
unsigned l = i << 1;
unsigned r = l + 1;
if(l < tasks.size() && r < tasks.size()) {
tasks[i].precede(tasks[l], tasks[r]);
}
}
executor.run(taskflow).get();
assert(counter + 1 == tasks.size());
}
std::chrono::microseconds measure_time_taskflow(
size_t num_layers,
unsigned num_threads
) {
auto beg = std::chrono::high_resolution_clock::now();
binary_tree_taskflow(num_layers, num_threads);
auto end = std::chrono::high_resolution_clock::now();
return std::chrono::duration_cast<std::chrono::microseconds>(end - beg);
}