Use realloc for histogram cache and expose the cache limit. (#9455)

2023-08-10 14:05:27 +08:00
parent a57371ef7c
commit 1caa93221a
10 changed files with 71 additions and 20 deletions
--- a/tests/cpp/tree/test_evaluate_splits.h
+++ b/tests/cpp/tree/test_evaluate_splits.h
@@ -59,7 +59,7 @@ class TestPartitionBasedSplit : public ::testing::Test {
    cuts_.min_vals_.Resize(1);

    HistMakerTrainParam hist_param;
-    hist_.Reset(cuts_.TotalBins(), hist_param.internal_max_cached_hist_node);
+    hist_.Reset(cuts_.TotalBins(), hist_param.max_cached_hist_node);
    hist_.AllocateHistograms({0});
    auto node_hist = hist_[0];