Function TestSparseDMatrixLoad

tests/cpp/data/test_sparse_page_dmatrix.cc:21–56

using namespace xgboost;  // NOLINT
template <typename Page>
void TestSparseDMatrixLoad(Context const *ctx) {
  auto m = RandomDataGenerator{1024, 5, 0.0}.Batches(4).GenerateSparsePageDMatrix("temp", true);

  auto n_threads = 0;
  auto config = ExtMemConfig{"temp",
                             false,
                             ::xgboost::cuda_impl::AutoHostRatio(),
                             cuda_impl::MatchingPageBytes(),
                             std::numeric_limits<float>::quiet_NaN(),
                             n_threads};
  ASSERT_EQ(AllThreadsForTest(), m->Ctx()->Threads());
  ASSERT_EQ(m->Info().num_col_, 5);
  ASSERT_EQ(m->Info().num_row_, 1024);

  auto simple = RandomDataGenerator{1024, 5, 0.0}.GenerateDMatrix(true);
  Page out;
  for (auto const &page : m->GetBatches<Page>(ctx)) {
    if (std::is_same_v<Page, SparsePage>) {
      out.Push(page);
    } else {
      out.PushCSC(page);
    }
  }
  ASSERT_EQ(m->Info().num_col_, simple->Info().num_col_);
  ASSERT_EQ(m->Info().num_row_, simple->Info().num_row_);
  if (std::is_same_v<Page, SortedCSCPage>) {
    out.SortRows(ctx->Threads());
  }

  for (auto const &page : simple->GetBatches<Page>(ctx)) {
    ASSERT_EQ(page.offset.HostVector(), out.offset.HostVector());
    for (size_t i = 0; i < page.data.Size(); ++i) {
      ASSERT_EQ(page.data.HostVector()[i].fvalue, out.data.HostVector()[i].fvalue);
    }
  }
}

TEST(SparsePageDMatrix, Load) {
  Context ctx;