Support column split in approx tree method (#8847)

2023-03-01 11:59:07 -08:00
parent 6d8afb2218
commit 7cbaee9916
6 changed files with 101 additions and 19 deletions
--- a/src/tree/hist/histogram.h
+++ b/src/tree/hist/histogram.h
@@ -98,7 +98,7 @@ class HistogramBuilder {
                   std::vector<ExpandEntry> const &nodes_for_explicit_hist_build,
                   std::vector<ExpandEntry> const &nodes_for_subtraction_trick,
                   RegTree const *p_tree) {
-    if (is_distributed_) {
+    if (is_distributed_ && !is_col_split_) {
      this->AddHistRowsDistributed(starting_index, sync_count, nodes_for_explicit_hist_build,
                                   nodes_for_subtraction_trick, p_tree);
    } else {