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/python-package/xgboost/testing/params.py
+++ b/python-package/xgboost/testing/params.py
@@ -42,7 +42,7 @@ hist_parameter_strategy = strategies.fixed_dictionaries(
 )

 hist_cache_strategy = strategies.fixed_dictionaries(
-    {"internal_max_cached_hist_node": strategies.sampled_from([1, 4, 1024, 2**31])}
+    {"max_cached_hist_node": strategies.sampled_from([1, 4, 1024, 2**31])}
 )

 hist_multi_parameter_strategy = strategies.fixed_dictionaries(