xgboost 0.72版本
将xgboost的CMakeList.txt里的openmp相关的删了
加锁
pthread_mutex_lock(&lock_);
XGBoosterPredict(h_booster, h_test, 0, 0, &out_len, &f);
pthread_mutex_unlock(&lock_);
给predict加锁,又屏蔽了openmp,相当于xgboost部分是单线程的,其他啥的是多线程,XGBoosterPredict本身速度很快
微信扫一扫
xgboost 0.72版本
将xgboost的CMakeList.txt里的openmp相关的删了
加锁
pthread_mutex_lock(&lock_);
XGBoosterPredict(h_booster, h_test, 0, 0, &out_len, &f);
pthread_mutex_unlock(&lock_);
给predict加锁,又屏蔽了openmp,相当于xgboost部分是单线程的,其他啥的是多线程,XGBoosterPredict本身速度很快
相关推荐