[EM] Return a full DMatrix instead of a Ellpack from the GPU sampler. (#10753)

This commit is contained in:
Jiaming Yuan
2024-08-28 01:05:11 +08:00
committed by GitHub
parent d6ebcfb032
commit bde1265caf
20 changed files with 525 additions and 214 deletions

View File

@@ -110,8 +110,15 @@ class MetaInfo {
* @brief Validate all metainfo.
*/
void Validate(DeviceOrd device) const;
MetaInfo Slice(common::Span<int32_t const> ridxs) const;
/**
* @brief Slice the meta info.
*
* The device of ridxs is specified by the ctx object.
*
* @param ridxs Index of selected rows.
* @param nnz The number of non-missing values.
*/
MetaInfo Slice(Context const* ctx, common::Span<bst_idx_t const> ridxs, bst_idx_t nnz) const;
MetaInfo Copy() const;
/**