[EM] Refactor GPU histogram builder. (#10764)
- 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:
@@ -522,6 +522,7 @@ XGB_DLL int XGQuantileDMatrixCreateFromCallback(DataIterHandle iter, DMatrixHand
|
||||
* - nthread (optional): Number of threads used for initializing DMatrix.
|
||||
* - max_bin (optional): Maximum number of bins for building histogram. Must be consistent with
|
||||
the corresponding booster training parameter.
|
||||
* - on_host (optional): Whether the data should be placed on host memory. Used by GPU inputs.
|
||||
* @param out The created Quantile DMatrix.
|
||||
*
|
||||
* @return 0 when success, -1 when failure happens
|
||||
|
||||
Reference in New Issue
Block a user