threads.h

Go to the documentation of this file.

 
#pragma once
 
#include <stdexcept>
 
#ifdef _OPENMP
#include <omp.h>
#endif
 
namespace slope {
 
class Threads
{
public:
  static void set(const int 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 int get() { return num_threads; }
 
private:
#ifdef _OPENMP
  inline static int num_threads = std::max(1, omp_get_max_threads() / 2);
#else
  inline static int num_threads = 1;
#endif
};
 
} // namespace slope