backend_openblas.hpp

Go to the documentation of this file.

#pragma once
 
#include <uni20/config.hpp>
 
#include <string>
 
#if !UNI20_BACKEND_OPENBLAS
#error "backend_openblas.hpp requires UNI20_BACKEND_OPENBLAS"
#endif
 
namespace uni20
{
namespace blas::openblas
{
namespace detail
{
extern "C"
{
  void openblas_set_num_threads(int num_threads);
  int openblas_get_num_threads();
  int openblas_get_num_procs();
  char* openblas_get_config();
  char* openblas_get_corename();
  int openblas_get_parallel();
}
 
inline std::string copy_string(char const* value) { return value == nullptr ? std::string{} : std::string(value); }
} // namespace detail
 
enum class ParallelMode : int
{
  sequential = 0,
  thread = 1,
  openmp = 2
};
 
inline std::string config() { return detail::copy_string(detail::openblas_get_config()); }
 
inline std::string core_name() { return detail::copy_string(detail::openblas_get_corename()); }
 
inline void set_num_threads(int threads) { detail::openblas_set_num_threads(threads); }
 
inline int num_threads() { return detail::openblas_get_num_threads(); }
 
inline int num_procs() { return detail::openblas_get_num_procs(); }
 
inline ParallelMode parallel_mode() { return static_cast<ParallelMode>(detail::openblas_get_parallel()); }
} // namespace blas::openblas
} // namespace uni20