compatibility issue with openmp

2014-03-03 15:11:41 -08:00
parent e3b7abfb47
commit 2aa1978cb6
6 changed files with 24 additions and 5 deletions
--- a/booster/tree/xgboost_col_treemaker.hpp
+++ b/booster/tree/xgboost_col_treemaker.hpp
@@ -7,9 +7,9 @@
 * \author Tianqi Chen: tianqi.tchen@gmail.com 
 */
 // use openmp
-#include <omp.h>
 #include <vector>
 #include "xgboost_tree_model.h"
+#include "../../utils/xgboost_omp.h"
 #include "../../utils/xgboost_random.h"

 namespace xgboost{
--- a/booster/xgboost_gbmbase.h
+++ b/booster/xgboost_gbmbase.h
@@ -1,9 +1,9 @@
 #ifndef XGBOOST_GBMBASE_H
 #define XGBOOST_GBMBASE_H

-#include <omp.h>
 #include <cstring>
 #include "xgboost.h"
+#include "../utils/xgboost_omp.h"
 #include "../utils/xgboost_config.h"
 /*!
 * \file xgboost_gbmbase.h