gmOmp.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_OPENMP_H_
#define _GEMA_OPENMP_H_

#ifdef _OPENMP
#include <omp.h>
#endif

inline void GmOmpInit(int maxThreads)
{
  assert(maxThreads > 0);
#ifdef _OPENMP
  omp_set_num_threads(maxThreads);
  omp_set_dynamic(0);
#endif
}

inline int GmOmpAdjustNumThreads(int nt) 
{ 
#ifdef _OPENMP
  int mt = omp_get_max_threads();
  assert(mt > 0);
  return (nt > 0 && nt <= mt) ? nt : mt;
#else
  return 1;
#endif
}

inline int GmOmpThreadNum()
{
#ifdef _OPENMP
  return omp_get_thread_num();
#else
  return 0;
#endif
}

#ifdef _OPENMP
#define GM_OMP_ASSERT_NO_DYNAMIC assert(omp_get_dynamic() == 0)
#else
#define GM_OMP_ASSERT_NO_DYNAMIC ((void)0)
#endif

#endif