Add data split mode to DMatrix MetaInfo (#8568)

2022-12-25 04:37:37 -08:00
parent 77b069c25d
commit 3ceeb8c61c
20 changed files with 113 additions and 103 deletions
--- a/src/data/simple_dmatrix.cc
+++ b/src/data/simple_dmatrix.cc
@@ -65,6 +65,7 @@ DMatrix* SimpleDMatrix::SliceCol(std::size_t start, std::size_t size) {
    out->Info() = this->Info().Copy();
    out->Info().num_nonzero_ = h_offset.back();
  }
+  out->Info().data_split_mode = DataSplitMode::kCol;
  return out;
 }