parallel.hpp

Go to the documentation of this file.

#ifndef IRLBA_PARALLEL_HPP
#define IRLBA_PARALLEL_HPP
 
#include "utils.hpp"
#include <vector>
#include "Eigen/Dense"
 
#ifndef IRLBA_CUSTOM_PARALLEL
#include "subpar/subpar.hpp"
#endif
 
namespace irlba {
 
template<typename Task_, class Run_>
void parallelize(Task_ num_tasks, Run_ run_task) {
#ifndef IRLBA_CUSTOM_PARALLEL
    // Use cases here don't allocate or throw, so nothrow_ = true is fine.
    subpar::parallelize_simple<true>(num_tasks, std::move(run_task));
#else
    IRLBA_CUSTOM_PARALLEL(num_tasks, run_task);
#endif
}
 
class EigenThreadScope {
#ifndef _OPENMP
public:
    EigenThreadScope([[maybe_unused]] int num_threads) {}
 
#else
public:
    EigenThreadScope([[maybe_unused]] int num_threads) : my_previous(Eigen::nbThreads()) {
#ifdef IRLBA_CUSTOM_PARALLEL
#ifdef IRLBA_CUSTOM_PARALLEL_USES_OPENMP
        Eigen::setNbThreads(num_threads);
#else
        Eigen::setNbThreads(1);
#endif
#else
#ifdef SUBPAR_USES_OPENMP_SIMPLE
        Eigen::setNbThreads(num_threads);
#else
        Eigen::setNbThreads(1);
#endif
#endif
    }
 
private:
    int my_previous;
 
public:
    ~EigenThreadScope() { 
        Eigen::setNbThreads(my_previous);
    }
#endif
 
public:
    EigenThreadScope(const EigenThreadScope&) = delete;
    EigenThreadScope(EigenThreadScope&&) = delete;
    EigenThreadScope& operator=(const EigenThreadScope&) = delete;
    EigenThreadScope& operator=(EigenThreadScope&&) = delete;
};
 
}
 
#endif