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/helpers.h
+++ b/tests/cpp/helpers.h
@@ -154,7 +154,7 @@ class SimpleRealUniformDistribution {
 std::shared_ptr<xgboost::DMatrix> *CreateDMatrix(int rows, int columns,
                                                 float sparsity, int seed = 0);

-std::unique_ptr<DMatrix> CreateSparsePageDMatrix();
+std::unique_ptr<DMatrix> CreateSparsePageDMatrix(size_t n_entries, size_t page_size);

 gbm::GBTreeModel CreateTestModel();