backend_mkl.hpp

Go to the documentation of this file.

#pragma once
 
#include <uni20/config.hpp>
 
#include <mkl.h>
#include <string>
 
#if !UNI20_BACKEND_MKL
#error "backend_mkl.hpp requires UNI20_BACKEND_MKL"
#endif
 
namespace uni20
{
namespace blas::mkl
{
constexpr bool sequential_backend() { return UNI20_BACKEND_MKL_SEQUENTIAL; }
 
constexpr bool threaded_backend() { return UNI20_BACKEND_MKL_THREADED; }
 
enum class Domain : int
{
  all = MKL_DOMAIN_ALL,
  blas = MKL_DOMAIN_BLAS,
  fft = MKL_DOMAIN_FFT,
  vml = MKL_DOMAIN_VML,
  pardiso = MKL_DOMAIN_PARDISO,
  lapack = MKL_DOMAIN_LAPACK
};
 
inline std::string version_string()
{
  char buffer[198]{};
  MKL_Get_Version_String(buffer, static_cast<int>(sizeof(buffer)));
  return std::string(buffer);
}
 
inline void set_num_threads(int threads) { MKL_Set_Num_Threads(threads); }
 
inline int max_threads() { return MKL_Get_Max_Threads(); }
 
inline int set_num_threads(int threads, Domain domain)
{
  return MKL_Domain_Set_Num_Threads(threads, static_cast<int>(domain));
}
 
inline int max_threads(Domain domain) { return MKL_Domain_Get_Max_Threads(static_cast<int>(domain)); }
} // namespace blas::mkl
} // namespace uni20