gmBlas.h

Go to the documentation of this file.

/************************************************************************
**
** Copyright (C) 2014 by Carlos Augusto Teixera Mendes
** All rights reserved.
**
** This file is part of the "GeMA" software. It's use should respect
** the terms in the license agreement that can be found together
** with this source code.
** It is provided AS IS, with NO WARRANTY OF ANY KIND,
** INCLUDING THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR
** A PARTICULAR PURPOSE.
**
************************************************************************/
 
#ifndef _GEMA_BLAS_H_
#define _GEMA_BLAS_H_
 
#include <assert.h>
 
#ifdef GEMA_OPEN_BLAS
 
extern "C"
{
  void openblas_set_num_threads(int num_threads);
  int  openblas_get_num_threads(void);
}
 
#define GEMA_BLAS_SET_NUM_THREADS(n) openblas_set_num_threads(n)
#define GEMA_BLAS_GET_NUM_THREADS()   openblas_get_num_threads()
 
#elif defined GEMA_MKL_BLAS
 
extern "C"
{
    void MKL_Set_Num_Threads(int num_threads);
    int  MKL_Get_Max_Threads(void);
}
 
 
#define GEMA_BLAS_SET_NUM_THREADS(n)  MKL_Set_Num_Threads(n)
#define GEMA_BLAS_GET_NUM_THREADS()   MKL_Get_Max_Threads()
 
#elif defined GEMA_AOCL_BLAS
 
extern "C"
{
    void bli_thread_set_num_threads(int num_threads);
    int  bli_thread_get_num_threads(void);
}
 
#define GEMA_BLAS_SET_NUM_THREADS(n)  bli_thread_set_num_threads(n)
#define GEMA_BLAS_GET_NUM_THREADS()   bli_thread_get_num_threads()
 
#else
 
#define GEMA_BLAS_SET_NUM_THREADS(n)  ((void)0)
#define GEMA_BLAS_GET_NUM_THREADS()   (1)
 
#endif
 
// GmBlasLibName
#if defined GEMA_MKL_BLAS
constexpr const char* GmBlasLibName = "Intel Mkl";
#elif defined GEMA_OPEN_BLAS
constexpr const char* GmBlasLibName = "OpenBlas";
#elif defined GEMA_AOCL_BLAS
constexpr const char* GmBlasLibName = "AMD AOCL";
#elif defined GEMA_STD_BLAS
constexpr const char* GmBlasLibName = "Generic";
#else
constexpr const char* GmBlasLibName = "Unknown";
#endif
 
// GmArmaBlasConfigFlag e GmArmaLapackConfigFlag  (either "no " or "")
#ifdef ARMA_DONT_USE_BLAS
constexpr const char* GmArmaBlasConfigFlag = "no ";
#else
constexpr const char* GmArmaBlasConfigFlag = "";
#endif
 
#ifdef ARMA_DONT_USE_LAPACK
constexpr const char* GmArmaLapackConfigFlag = "no ";
#else
constexpr const char* GmArmaLapackConfigFlag = "";
#endif
 
 
inline void GmBlasInit(int maxThreads)
{
  assert(maxThreads > 0);
  GEMA_BLAS_SET_NUM_THREADS(maxThreads);
 
}
 
class GmBlasTempUpdateNumThreads
{
public:
  GmBlasTempUpdateNumThreads(int nthreads) 
  { 
    if(nthreads)
    {
      _oldNumThreads = GEMA_BLAS_GET_NUM_THREADS();
      GEMA_BLAS_SET_NUM_THREADS(nthreads);
    }
    else
      _oldNumThreads = -1;
  }
 
  ~GmBlasTempUpdateNumThreads() 
  { 
    if(_oldNumThreads != -1) GEMA_BLAS_SET_NUM_THREADS(_oldNumThreads);
  }
private:
 
  int _oldNumThreads;
};
 
 
#endif