parallelize.hpp

Go to the documentation of this file.

#ifndef TATAMI_PYTHON_PARALLELIZE_HPP
#define TATAMI_PYTHON_PARALLELIZE_HPP
 
#ifdef TATAMI_PYTHON_PARALLELIZE_UNKNOWN
#include "pybind11/pybind11.h"
#include "subpar/subpar.hpp"
 
#include <optional>
 
#ifndef TATAMI_PYTHON_SERIALIZE
#define TATAMI_PYTHON_SERIALIZE ::tatami_python::lock
#endif 
 
namespace tatami_python {
 
template<class Function_, class Index_>
void parallelize(const Function_ fun, const Index_ tasks, int threads) {
    std::optional<pybind11::gil_scoped_release> ungil;
    if (PyGILState_Check()) {
        ungil.emplace();
    }
    subpar::parallelize_range(threads, tasks, std::move(fun));
}
 
template<typename Function_>
void lock(Function_ fun) {
    std::optional<pybind11::gil_scoped_acquire> gil;
    if (!PyGILState_Check()) {
        gil.emplace();
    }
    fun();
}
 
}
 
#endif
#endif