Sort sparse page index when constructing DMatrix. (#7731)

2022-03-16 18:01:05 +08:00
parent 613ec36c5a
commit e78a38b837
4 changed files with 67 additions and 1 deletions
--- a/src/data/simple_dmatrix.cc
+++ b/src/data/simple_dmatrix.cc
@@ -211,6 +211,11 @@ SimpleDMatrix::SimpleDMatrix(AdapterT* adapter, float missing, int nthread) {
    info_.num_row_ = adapter->NumRows();
  }
  info_.num_nonzero_ = data_vec.size();
+
+  // Sort the index for row partitioners used by variuos tree methods.
+  if (!sparse_page_->IsIndicesSorted(this->ctx_.Threads())) {
+    sparse_page_->SortIndices(this->ctx_.Threads());
+  }
 }

 SimpleDMatrix::SimpleDMatrix(dmlc::Stream* in_stream) {