fix wrapper checkNAN

This commit is contained in:
tqchen 2015-03-08 09:52:59 -07:00
parent d202d8b977
commit e79840e620
2 changed files with 4 additions and 3 deletions

View File

@ -28,12 +28,13 @@ extern "C" {
void (*Check)(int exp, const char *fmt, ...) = XGBoostCheck_R;
void (*Error)(const char *fmt, ...) = error;
}
} // namespace utils
namespace wrapper {
bool CheckNAN(float v) {
return ISNAN(v);
}
}
} // namespace utils
} // namespace wrapper
namespace random {
void Seed(unsigned seed) {

View File

@ -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)