Support column split in multi-target hist (#9171)

2023-05-26 01:56:05 -07:00
parent acd363033e
commit 5b69534b43
17 changed files with 386 additions and 96 deletions
--- a/src/common/quantile.h
+++ b/src/common/quantile.h
@@ -822,7 +822,8 @@ class SketchContainerImpl {
    return group_ind;
  }
  // Gather sketches from all workers.
-  void GatherSketchInfo(std::vector<typename WQSketch::SummaryContainer> const &reduced,
+  void GatherSketchInfo(MetaInfo const& info,
+                        std::vector<typename WQSketch::SummaryContainer> const &reduced,
                        std::vector<bst_row_t> *p_worker_segments,
                        std::vector<bst_row_t> *p_sketches_scan,
                        std::vector<typename WQSketch::Entry> *p_global_sketches);