threads.h

Go to the documentation of this file.

 
#pragma once
 
#include <stdexcept>
 
#ifdef _OPENMP
#include <omp.h>
#endif
 
namespace qualpal {
 
class Threads
{
public:
  static void set(const std::size_t n)
  {
    if (n > 0) {
      num_threads = n;
#ifdef _OPENMP
      omp_set_num_threads(n);
#endif
    } else {
      throw std::invalid_argument("Number of threads must be positive");
    }
  }
 
  static std::size_t get() { return num_threads; }
 
private:
#ifdef _OPENMP
  inline static std::size_t num_threads =
    std::max(1, omp_get_max_threads() / 2);
#else
  inline static std::size_t num_threads = 1;
#endif
};
 
} // namespace qualpal