Use ellpack for prediction only when sparsepage doesn't exist. (#5504)

2020-04-10 12:15:46 +08:00
parent ad826e913f
commit 6671b42dd4
35 changed files with 166 additions and 116 deletions
--- a/tests/cpp/histogram_helpers.h
+++ b/tests/cpp/histogram_helpers.h
@@ -22,7 +22,7 @@ class HistogramCutsWrapper : public common::HistogramCuts {

 inline std::unique_ptr<EllpackPageImpl> BuildEllpackPage(
    int n_rows, int n_cols, bst_float sparsity= 0) {
-  auto dmat = RandomDataGenerator(n_rows, n_cols, sparsity).Seed(3).GenerateDMatix();
+  auto dmat = RandomDataGenerator(n_rows, n_cols, sparsity).Seed(3).GenerateDMatrix();
  const SparsePage& batch = *dmat->GetBatches<xgboost::SparsePage>().begin();

  HistogramCutsWrapper cmat;