Support column split in histogram builder (#8811)

2023-02-17 06:37:01 -08:00
parent 40fd3d6d5f
commit a65ad0bd9c
8 changed files with 38 additions and 22 deletions
--- a/src/data/iterative_dmatrix.cc
+++ b/src/data/iterative_dmatrix.cc
@@ -213,7 +213,7 @@ void IterativeDMatrix::InitFromCPU(DataIterHandle iter_handle, float missing,
        SyncFeatureType(&h_ft);
        p_sketch.reset(new common::HostSketchContainer{
            batch_param_.max_bin, h_ft, column_sizes, !proxy->Info().group_ptr_.empty(),
-            proxy->Info().data_split_mode == DataSplitMode::kCol, ctx_.Threads()});
+            proxy->IsColumnSplit(), ctx_.Threads()});
      }
      HostAdapterDispatch(proxy, [&](auto const& batch) {
        proxy->Info().num_nonzero_ = batch_nnz[i];