Support cpu quantile sketch with column-wise data split (#8742)

This commit is contained in:
Rong Ou
2023-02-04 22:26:24 -08:00
committed by GitHub
parent c1786849e3
commit 66191e9926
15 changed files with 250 additions and 118 deletions

View File

@@ -627,11 +627,11 @@ class DMatrix {
/**
* \brief Slice a DMatrix by columns.
*
* @param start The position of the first column
* @param size The number of columns in the slice
* @param num_slices Total number of slices
* @param slice_id Index of the current slice
* @return DMatrix containing the slice of columns
*/
virtual DMatrix *SliceCol(std::size_t start, std::size_t size) = 0;
virtual DMatrix *SliceCol(int num_slices, int slice_id) = 0;
protected:
virtual BatchSet<SparsePage> GetRowBatches() = 0;