Support column split with GPU quantile (#9370)

2023-07-10 21:15:56 -07:00
parent 97ed944209
commit 3632242e0b
6 changed files with 68 additions and 16 deletions
--- a/src/common/quantile.cuh
+++ b/src/common/quantile.cuh
@@ -154,9 +154,9 @@ class SketchContainer {
             Span<SketchEntry const> that);

  /* \brief Merge quantiles from other GPU workers. */
-  void AllReduce();
+  void AllReduce(bool is_column_split);
  /* \brief Create the final histogram cut values. */
-  void MakeCuts(HistogramCuts* cuts);
+  void MakeCuts(HistogramCuts* cuts, bool is_column_split);

  Span<SketchEntry const> Data() const {
    return {this->Current().data().get(), this->Current().size()};