Avoid thread block with sparse data. (#7255)

This commit is contained in:
Jiaming Yuan
2021-09-25 13:11:34 +08:00
committed by GitHub
parent ca17f8a5fc
commit d8a549e6ac
6 changed files with 97 additions and 16 deletions

View File

@@ -256,5 +256,10 @@ TEST(GPUPredictor, PredictLeafBasic) {
ASSERT_EQ(v, 0);
}
}
TEST(GPUPredictor, Sparse) {
TestSparsePrediction(0.2, "gpu_predictor");
TestSparsePrediction(0.8, "gpu_predictor");
}
} // namespace predictor
} // namespace xgboost