33 lines
912 B
C++
33 lines
912 B
C++
|
#include "reduce_sum.hpp"
|
||
|
|
#include <tbb/parallel_reduce.h>
|
||
|
|
#include <tbb/blocked_range.h>
|
||
|
|
#include <tbb/global_control.h>
|
||
|
|
|
||
|
|
// reduce_sum_tbb
|
||
|
|
void reduce_sum_tbb(unsigned num_threads) {
|
||
|
|
|
||
|
|
tbb::global_control c(
|
||
|
|
tbb::global_control::max_allowed_parallelism, num_threads
|
||
|
|
);
|
||
|
|
|
||
|
|
tbb::parallel_reduce(
|
||
|
|
tbb::blocked_range<double*>(vec.data(), vec.data() + vec.size()),
|
||
|
|
0.0,
|
||
|
|
[](const tbb::blocked_range<double*>& r, double value) {
|
||
|
|
return std::accumulate(r.begin(), r.end(), value);
|
||
|
|
},
|
||
|
|
[](double l, double r) -> double {
|
||
|
|
return l + r;
|
||
|
|
}
|
||
|
|
);
|
||
|
|
|
||
|
|
//std::cout << reduce_sum() << std::endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
std::chrono::microseconds measure_time_tbb(unsigned num_threads) {
|
||
|
|
auto beg = std::chrono::high_resolution_clock::now();
|
||
|
|
reduce_sum_tbb(num_threads);
|
||
|
|
auto end = std::chrono::high_resolution_clock::now();
|
||
|
|
return std::chrono::duration_cast<std::chrono::microseconds>(end - beg);
|
||
|
|
}
|