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 CPP API
阅读 45
2022-07-27
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本身速度很快
相关推荐
精彩评论(0)