Unify the code path between local and distributed training. (#9433)

This removes the need for a local histogram space during distributed training, which cuts the cache size by half.
This commit is contained in:
Jiaming Yuan
2023-08-03 21:46:36 +08:00
committed by GitHub
parent f958e32683
commit 1332ff787f
3 changed files with 45 additions and 173 deletions

View File

@@ -453,6 +453,7 @@ class HistCollection {
data_[0].resize(new_size);
}
}
[[nodiscard]] bool IsContiguous() const { return contiguous_allocation_; }
private:
/*! \brief number of all bins over all features */