Initial support for external memory in gpu_predictor (#4284)

2019-05-02 18:01:27 -07:00
parent 54980b8959
commit feb6ae3e18
11 changed files with 73 additions and 15 deletions
--- a/tests/cpp/data/test_sparse_page_dmatrix.cc
+++ b/tests/cpp/data/test_sparse_page_dmatrix.cc
@@ -26,7 +26,7 @@ TEST(SparsePageDMatrix, MetaInfo) {
 }

 TEST(SparsePageDMatrix, RowAccess) {
-  std::unique_ptr<xgboost::DMatrix> dmat = xgboost::CreateSparsePageDMatrix();
+  std::unique_ptr<xgboost::DMatrix> dmat = xgboost::CreateSparsePageDMatrix(12, 64);

  // Test the data read into the first row
  auto &batch = *dmat->GetRowBatches().begin();