Morpheus_OpenMPUtils.hpp

 
#ifndef MORPHEUS_OPENMP_UTILS_HPP
#define MORPHEUS_OPENMP_UTILS_HPP
 
#include <Morpheus_Macros.hpp>
#if defined(MORPHEUS_ENABLE_OPENMP)
 
#include <omp.h>
 
namespace Morpheus {
namespace Impl {
 
template <typename T>
T _split_work(const T load, const T workers, const T worker_id) {
  const T unifload = load / workers;  // uniform distribution
  const T rem      = load - unifload * workers;
  T bound;
 
  //  round-robin assignment of the remaining work
  if (worker_id <= rem) {
    bound = (unifload + 1) * worker_id;
  } else {
    bound = (unifload + 1) * rem + unifload * (worker_id - rem);
  }
 
  return bound;
}
 
template <typename T>
void atomic_add(T* out, T val) {
#pragma omp atomic
  *out += val;
}
 
template <typename T = int>
T threads() {
  T t = 1;
#pragma omp parallel
  { t = omp_get_num_threads(); }
 
  return t;
}
 
}  // namespace Impl
}  // namespace Morpheus
 
#endif  // MORPHEUS_ENABLE_OPENMP
#endif  // MORPHEUS_OPENMP_UTILS_HPP