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

[Breaking] Remove rabit support for custom reductions and grow_local_histmaker updater (#7992)

Browse Source
This commit is contained in:
Rong Ou
2022-06-21 00:08:23 -07:00
committed by GitHub
parent 4a87ea49b8
commit e5ec546da5
17 changed files with 36 additions and 1100 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/tree/hist/histogram.h
View File

@@ -24,7 +24,6 @@ class HistogramBuilder {
common::HistCollection hist_local_worker_;
common::GHistBuilder builder_;
common::ParallelGHistBuilder buffer_;
rabit::Reducer<GradientPairPrecise, GradientPairPrecise::Reduce> reducer_;
BatchParam param_;
int32_t n_threads_{-1};
size_t n_batches_{0};
@@ -199,8 +198,8 @@ class HistogramBuilder {
}
});
reducer_.Allreduce(this->hist_[starting_index].data(),
builder_.GetNumBins() * sync_count);
rabit::Allreduce<rabit::op::Sum>(reinterpret_cast<double*>(this->hist_[starting_index].data()),
builder_.GetNumBins() * sync_count * 2);
ParallelSubtractionHist(space, nodes_for_explicit_hist_build,
nodes_for_subtraction_trick, p_tree);