Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix gpu_hist apply_split test. (#4158)

Browse Source
This commit is contained in:
Jiaming Yuan
2019-02-18 02:48:28 +08:00
committed by GitHub
parent 2e618af743
commit e1240413c9
3 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/cpp/predictor/test_gpu_predictor.cu
View File

@@ -90,6 +90,7 @@ TEST(gpu_predictor, Test) {
delete dmat;
}
#if defined(XGBOOST_USE_NCCL)
// Test whether pickling preserves predictor parameters
TEST(gpu_predictor, MGPU_PicklingTest) {
int ngpu;
@@ -163,7 +164,9 @@ TEST(gpu_predictor, MGPU_PicklingTest) {
CheckCAPICall(XGBoosterFree(bst2));
}
#endif // defined(XGBOOST_USE_NCCL)
#if defined(XGBOOST_USE_NCCL)
// multi-GPU predictor test
TEST(gpu_predictor, MGPU_Test) {
std::unique_ptr<Predictor> gpu_predictor =
@@ -202,6 +205,6 @@ TEST(gpu_predictor, MGPU_Test) {
delete dmat;
}
}
#endif // defined(XGBOOST_USE_NCCL)
} // namespace predictor
} // namespace xgboost