4d81c741e9
* Generate column matrix from gHistIndex. * Avoid synchronization with the sparse page once the cache is written. * Cleanups: Remove member variables/functions, change the update routine to look like approx and gpu_hist. * Remove pruner.
33 lines
992 B
Plaintext
33 lines
992 B
Plaintext
/*!
|
|
* Copyright 2019-2022 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()) {
|
|
if (count_ != 0 && !sync_) {
|
|
// source is initialized to be the 0th page during construction, so when count_ is 0
|
|
// there's no need to increment the source.
|
|
++(*source_);
|
|
}
|
|
// This is not read from cache so we still need it to be synced with sparse page source.
|
|
CHECK_EQ(count_, source_->Iter());
|
|
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
|