17 lines
449 B
C
17 lines
449 B
C
#ifndef XGBOOST_UTILS_OMP_H_
|
|
#define XGBOOST_UTILS_OMP_H_
|
|
/*!
|
|
* \file omp.h
|
|
* \brief header to handle OpenMP compatibility issues
|
|
* \author Tianqi Chen
|
|
*/
|
|
#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 // XGBOOST_UTILS_OMP_H_
|