50 lines
1.1 KiB
C++
50 lines
1.1 KiB
C++
#include "linear_chain.hpp"
|
|
#include <tbb/global_control.h>
|
|
#include <tbb/flow_graph.h>
|
|
|
|
// linear_chain_tbb
|
|
void linear_chain_tbb(size_t length, unsigned num_threads) {
|
|
|
|
using namespace tbb;
|
|
using namespace tbb::flow;
|
|
|
|
size_t counter = 0;
|
|
|
|
tbb::global_control c(
|
|
tbb::global_control::max_allowed_parallelism, num_threads
|
|
);
|
|
|
|
graph g;
|
|
|
|
std::vector<continue_node<continue_msg>*> tasks(length);
|
|
|
|
for(size_t i=0; i<tasks.size(); i++) {
|
|
tasks[i] = new continue_node<continue_msg>(g,
|
|
[&]( const continue_msg& ) {
|
|
counter++;
|
|
}
|
|
);
|
|
if(i) {
|
|
make_edge(*tasks[i-1], *tasks[i]);
|
|
}
|
|
}
|
|
|
|
tasks[0]->try_put(continue_msg());
|
|
g.wait_for_all();
|
|
|
|
for(auto& task : tasks) {
|
|
delete task;
|
|
}
|
|
|
|
assert(counter == tasks.size());
|
|
}
|
|
|
|
std::chrono::microseconds measure_time_tbb(
|
|
size_t length,
|
|
unsigned num_threads
|
|
) {
|
|
auto beg = std::chrono::high_resolution_clock::now();
|
|
linear_chain_tbb(length, num_threads);
|
|
auto end = std::chrono::high_resolution_clock::now();
|
|
return std::chrono::duration_cast<std::chrono::microseconds>(end - beg);
|
|
}
|