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.
27 lines
865 B
C++
27 lines
865 B
C++
/*!
|
|
* Copyright 2021-2022 by XGBoost Contributors
|
|
*/
|
|
#include "gradient_index_page_source.h"
|
|
|
|
namespace xgboost {
|
|
namespace data {
|
|
void GradientIndexPageSource::Fetch() {
|
|
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 GHistIndexMatrix());
|
|
CHECK_NE(cuts_.Values().size(), 0);
|
|
this->page_->Init(*csr, feature_types_, cuts_, max_bin_per_feat_, is_dense_, sparse_thresh_,
|
|
nthreads_);
|
|
this->WriteCache();
|
|
}
|
|
}
|
|
} // namespace data
|
|
} // namespace xgboost
|