Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[EM] Refactor GPU histogram builder. (#10764)

Browse Source 
- Expose the maximum number of cached nodes to be consistent with the CPU implementation. Also easier for testing.
- Extract the subtraction trick for easier testing.
- Split up the `GradientQuantiser` to avoid circular dependency.
This commit is contained in:
Jiaming Yuan
2024-08-30 02:39:14 +08:00
committed by GitHub
parent 34937fea41
commit 61dd854a52
17 changed files with 394 additions and 187 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
doc/parameter.rst
View File

@@ -232,12 +232,12 @@ Parameters for Tree Booster
* ``max_cached_hist_node``, [default = 65536]
Maximum number of cached nodes for CPU histogram.
Maximum number of cached nodes for histogram.
.. versionadded:: 2.0.0
- For most of the cases this parameter should not be set except for growing deep trees
on CPU.
- For most of the cases this parameter should not be set except for growing deep
trees. After 3.0, this parameter affects GPU algorithms as well.
.. _cat-param: