e6088366df
* Add Python iterator interface. * Add tests. * Add demo. * Add documents. * Handle empty dataset.
26 lines
660 B
Plaintext
26 lines
660 B
Plaintext
/*!
|
|
* Copyright 2019-2021 XGBoost contributors
|
|
*/
|
|
#include <memory>
|
|
#include <utility>
|
|
|
|
#include "ellpack_page.cuh"
|
|
#include "ellpack_page_source.h"
|
|
|
|
namespace xgboost {
|
|
namespace data {
|
|
void EllpackPageSource::Fetch() {
|
|
dh::safe_cuda(cudaSetDevice(param_.gpu_id));
|
|
if (!this->ReadCache()) {
|
|
auto const &csr = source_->Page();
|
|
this->page_.reset(new EllpackPage{});
|
|
auto *impl = this->page_->Impl();
|
|
*impl = EllpackPageImpl(param_.gpu_id, *cuts_, *csr, is_dense_, row_stride_,
|
|
feature_types_);
|
|
page_->SetBaseRowId(csr->base_rowid);
|
|
this->WriteCache();
|
|
}
|
|
}
|
|
} // namespace data
|
|
} // namespace xgboost
|