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

[EM] Add GPU version of the external memory QDM. (#10689)

Browse Source
This commit is contained in:
Jiaming Yuan
2024-08-10 10:49:43 +08:00
committed by GitHub
parent 18b28d9315
commit d414fdf2e7
25 changed files with 536 additions and 261 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/xgboost/data.h
View File

@@ -641,7 +641,7 @@ class DMatrix {
typename XGDMatrixCallbackNext>
static DMatrix* Create(DataIterHandle iter, DMatrixHandle proxy, std::shared_ptr<DMatrix> ref,
DataIterResetCallback* reset, XGDMatrixCallbackNext* next, float missing,
std::int32_t nthread, bst_bin_t max_bin, std::string cache);
std::int32_t nthread, bst_bin_t max_bin, std::string cache, bool on_host);
virtual DMatrix *Slice(common::Span<int32_t const> ridxs) = 0;