cpp-taskflow/benchmarks/linear_pipeline/omp.cpp at master · feiyunwill/cpp-taskflow · GitHub

49 lines (40 loc) · 1010 Bytes

#include "linear_pipeline.hpp"

#include <omp.h>

// linear_chain_omp

void linear_chain_omp(size_t length, unsigned num_threads) {

size_t counter = 0;

size_t *D = new size_t [length];

#pragma omp parallel num_threads(num_threads)

{

#pragma omp single

{

for(size_t i=0; i<length; ++i) {

if(i==0) {

#pragma omp task firstprivate(i) depend(out:D[i])

{

++counter;

}

else {

#pragma omp task firstprivate(i) depend(out:D[i]) depend(in:D[i-1])

{

++counter;

}

assert(counter == length);

delete [] D;

}

std::chrono::microseconds measure_time_omp(

std::string,

unsigned length,

unsigned num_threads,

size_t

) {

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

linear_chain_omp(length, num_threads);

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

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

}

Expand file tree