Accept histogram cut instead gradient index in evaluation. (#7336)

2021-10-20 18:04:46 +08:00
parent 15685996fc
commit 8d7c6366d7
4 changed files with 13 additions and 11 deletions
--- a/tests/cpp/data/test_gradient_index.cc
+++ b/tests/cpp/data/test_gradient_index.cc
@@ -13,7 +13,8 @@ TEST(GradientIndex, ExternalMemory) {
  std::unique_ptr<DMatrix> dmat = CreateSparsePageDMatrix(10000);
  std::vector<size_t> base_rowids;
  std::vector<float> hessian(dmat->Info().num_row_, 1);
-  for (auto const& page : dmat->GetBatches<GHistIndexMatrix>({0, 64, hessian})) {
+  for (auto const &page : dmat->GetBatches<GHistIndexMatrix>(
+           {GenericParameter::kCpuId, 64, hessian})) {
    base_rowids.push_back(page.base_rowid);
  }
  size_t i = 0;