Support distributed CPU env for categorical data. (#7575)

* Add support for cat data in sketch allreduce. * Share tests between CPU and GPU.
2022-01-18 21:56:07 +08:00
parent deab0e32ba
commit cc06fab9a7
5 changed files with 299 additions and 150 deletions
--- a/tests/cpp/helpers.cc
+++ b/tests/cpp/helpers.cc
@@ -183,7 +183,7 @@ bool IsNear(std::vector<xgboost::bst_float>::const_iterator _beg1,
 }

 SimpleLCG::StateType SimpleLCG::operator()() {
-  state_ = (alpha_ * state_) % mod_;
+  state_ = (alpha_ * state_ + (state_ == 0 ? kDefaultInit : 0)) % mod_;
  return state_;
 }
 SimpleLCG::StateType SimpleLCG::Min() const { return min(); }