You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
470 B
27 lines
470 B
6 years ago
|
#ifndef COMMON_CPU
|
||
|
|
#define COMMON_CPU
|
||
|
|
|
||
|
|
#if defined(_OPENMP)
|
||
|
|
#include <omp.h>
|
||
|
|
#endif
|
||
|
|
|
||
|
|
template <typename FunctorT>
|
||
|
|
void iterate_cpu(FunctorT functor, int N) {
|
||
|
|
for(int idx = 0; idx < N; ++idx) {
|
||
|
|
functor(idx);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename FunctorT>
|
||
|
|
void iterate_omp_cpu(FunctorT functor, int N, int n_threads) {
|
||
|
|
#if defined(_OPENMP)
|
||
|
|
omp_set_num_threads(n_threads);
|
||
|
|
#pragma omp parallel for
|
||
|
|
#endif
|
||
|
|
for(int idx = 0; idx < N; ++idx) {
|
||
|
|
functor(idx);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|