Function TestSparsePageDMatrixDeterminism

tests/cpp/data/test_sparse_page_dmatrix.cc:345–368

}

auto TestSparsePageDMatrixDeterminism(std::int32_t n_threads) {
  std::vector<float> sparse_data;
  std::vector<size_t> sparse_rptr;
  std::vector<bst_feature_t> sparse_cids;

  common::TemporaryDirectory tmpdir;
  auto prefix = (tmpdir.Path() / "temp").string();
  auto dmat = RandomDataGenerator{4096, 64, 0.0}.Batches(4).GenerateSparsePageDMatrix(prefix, true);

  auto config = ExtMemConfig{prefix,
                             false,
                             ::xgboost::cuda_impl::AutoHostRatio(),
                             cuda_impl::MatchingPageBytes(),
                             std::numeric_limits<float>::quiet_NaN(),
                             n_threads};
  CHECK(dmat->Ctx()->Threads() == n_threads || dmat->Ctx()->Threads() == AllThreadsForTest());

  DMatrixToCSR(dmat.get(), &sparse_data, &sparse_rptr, &sparse_cids);

  auto cache_name =
      data::MakeId(prefix, dynamic_cast<data::SparsePageDMatrix *>(dmat.get())) + ".row.page";
  auto cache = common::LoadSequentialFile(cache_name);
  return cache;
}

TEST(SparsePageDMatrix, Determinism) {
#if defined(_MSC_VER)