Remove extra sync for dense data (#7120)

Co-authored-by: SHVETS, KIRILL <kirill.shvets@intel.com>
This commit is contained in:
ShvetsKS
2021-07-22 14:02:31 +03:00
committed by GitHub
parent e6088366df
commit caa9e527dd
2 changed files with 33 additions and 1 deletions

View File

@@ -329,8 +329,8 @@ void QuantileHistMaker::Builder<GradientSumT>::InitRoot(
for (const size_t *it = e.begin; it < e.end; ++it) {
grad_stat.Add(gpair_h[*it].GetGrad(), gpair_h[*it].GetHess());
}
histred_.Allreduce(&grad_stat, 1);
}
histred_.Allreduce(&grad_stat, 1);
auto weight = evaluator_->InitRoot(GradStats{grad_stat});
p_tree->Stat(RegTree::kRoot).sum_hess = grad_stat.GetHess();