Avoid thread block with sparse data. (#7255)

2021-09-25 13:11:34 +08:00
parent ca17f8a5fc
commit d8a549e6ac
6 changed files with 97 additions and 16 deletions
--- a/tests/cpp/predictor/test_predictor.h
+++ b/tests/cpp/predictor/test_predictor.h
@@ -70,6 +70,8 @@ void TestCategoricalPrediction(std::string name);
 void TestCategoricalPredictLeaf(StringView name);

 void TestIterationRange(std::string name);
+
+void TestSparsePrediction(float sparsity, std::string predictor);
 }  // namespace xgboost

 #endif  // XGBOOST_TEST_PREDICTOR_H_