fix wrapper checkNAN

2015-03-08 09:52:59 -07:00
parent d202d8b977
commit e79840e620
2 changed files with 4 additions and 3 deletions
--- a/src/tree/updater_colmaker-inl.hpp
+++ b/src/tree/updater_colmaker-inl.hpp
@@ -423,7 +423,7 @@ class ColMaker: public IUpdater {
      #endif
      int poption = param.parallel_option;
      if (poption == 2) {
-        poption = nsize * 2 < nthread ? 1 : 0;
+        poption = static_cast<int>(nsize) * 2 < nthread ? 1 : 0;
      }
      if (poption == 0) {
        #pragma omp parallel for schedule(dynamic, batch_size)