compatibility issue with openmp

This commit is contained in:
tqchen
2014-03-03 15:11:41 -08:00
parent 5a65f4b958
commit bbbbe6bc4e
6 changed files with 24 additions and 5 deletions

18
utils/xgboost_omp.h Normal file
View File

@@ -0,0 +1,18 @@
#ifndef XGBOOST_OMP_H
#define XGBOOST_OMP_H
/*!
* \file xgboost_omp.h
* \brief header to handle OpenMP compatibility issues
*
* \author Tianqi Chen: tianqi.tchen@gmail.com
*/
#if defined(_OPENMP)
#include <omp.h>
#else
#warning "OpenMP is not available, compile to single thread code"
inline int omp_get_thread_num() { return 0; }
inline int omp_get_num_threads() { return 1; }
inline void omp_set_num_threads( int nthread ) {}
#endif
#endif