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/include/xgboost/data.h
+++ b/include/xgboost/data.h
@@ -529,6 +529,11 @@ class DMatrix {
    return Info().num_nonzero_ == Info().num_row_ * Info().num_col_;
  }

+  /*! \brief Whether the data is split column-wise. */
+  bool IsColumnSplit() const {
+    return Info().data_split_mode == DataSplitMode::kCol;
+  }
+
  /*!
   * \brief Load DMatrix from URI.
   * \param uri The URI of input.