Support external memory in CPU histogram building. (#7372)

2021-11-23 01:13:33 +08:00
parent d33854af1b
commit 176110a22d
6 changed files with 305 additions and 162 deletions
--- a/tests/cpp/tree/hist/test_evaluate_splits.cc
+++ b/tests/cpp/tree/hist/test_evaluate_splits.cc
@@ -36,7 +36,7 @@ template <typename GradientSumT> void TestEvaluateSplits() {
  std::iota(row_indices.begin(), row_indices.end(), 0);
  row_set_collection.Init();

-  auto hist_builder = GHistBuilder<GradientSumT>(omp_get_max_threads(), gmat.cut.Ptrs().back());
+  auto hist_builder = GHistBuilder<GradientSumT>(gmat.cut.Ptrs().back());
  hist.Init(gmat.cut.Ptrs().back());
  hist.AddHistRow(0);
  hist.AllocateAllData();