taskflow/benchmarks/wavefront/taskflow.cpp at master · ModuleWorks/taskflow · GitHub

42 lines (32 loc) · 977 Bytes

#include "matrix.hpp"

#include <taskflow/taskflow.hpp>

// wavefront computing

void wavefront_taskflow(unsigned num_threads) {

tf::Executor executor(num_threads);

tf::Taskflow taskflow;

std::vector<std::vector<tf::Task>> node(MB);

for(auto &n : node){

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

n.emplace_back(taskflow.placeholder());

}

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

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

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

node[i][j].work(

[=]() {

block_computation(i, j);

}

);

if(j+1 < NB) node[i][j].precede(node[i][j+1]);

if(i+1 < MB) node[i][j].precede(node[i+1][j]);

}

executor.run(taskflow).get();

}

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

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

wavefront_taskflow(num_threads);

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

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

}

Expand file tree