Fix threads in DMatrix slice. (#8667)

2023-01-14 07:16:57 +08:00
parent e27cda7626
commit 07cf3d3e53
2 changed files with 24 additions and 6 deletions
--- a/src/data/simple_dmatrix.cc
+++ b/src/data/simple_dmatrix.cc
@@ -42,6 +42,7 @@ DMatrix* SimpleDMatrix::Slice(common::Span<int32_t const> ridxs) {
    out->Info() = this->Info().Slice(ridxs);
    out->Info().num_nonzero_ = h_offset.back();
  }
+  out->ctx_ = this->ctx_;
  return out;
 }