cpp-taskflow/benchmark/wavefront/tbb.cpp at cpp14 · walker2010cs/cpp-taskflow · GitHub

History

55 lines (41 loc) · 1.27 KB

Raw

#include "matrix.hpp"

#include <tbb/task_scheduler_init.h>

#include <tbb/flow_graph.h>

// the wavefront computation

void wavefront_tbb(unsigned num_threads) {

using namespace tbb;

using namespace tbb::flow;

tbb::task_scheduler_init init(num_threads);

continue_node<continue_msg> ***node = new continue_node<continue_msg> **[MB];

for ( int i = 0; i < MB; ++i ) {

node[i] = new continue_node<continue_msg> *[NB];

};

graph g;

matrix[M-1][N-1] = 0;

for( int i=MB; --i>=0; ) {

for( int j=NB; --j>=0; ) {

node[i][j] = new continue_node<continue_msg>( g,

[=]( const continue_msg& ) {

block_computation(i, j);

}

);

if(i+1 < MB) make_edge(*node[i][j], *node[i+1][j]);

if(j+1 < NB) make_edge(*node[i][j], *node[i][j+1]);

}

node[0][0]->try_put(continue_msg());

g.wait_for_all();

for(int i=0; i<MB; ++i) {

for(int j=0; j<NB; ++j) {

delete node[i][j];

}

delete [] node[i];

}

delete [] node;

}

std::chrono::microseconds measure_time_tbb(unsigned num_threads) {

auto beg = std::chrono::high_resolution_clock::now();

wavefront_tbb(num_threads);

auto end = std::chrono::high_resolution_clock::now();

return std::chrono::duration_cast<std::chrono::microseconds>(end - beg);

}

cpp-taskflow/benchmark/wavefront/tbb.cpp at cpp14 · walker2010cs/cpp-taskflow · GitHub

Expand file tree

Breadcrumbs

tbb.cpp

Latest commit

History

Breadcrumbs

tbb.cpp

File metadata and controls