Predict on Ellpack. (#5327)

* Unify GPU prediction node. * Add `PageExists`. * Dispatch prediction on input data for GPU Predictor.
2020-02-23 06:27:03 +08:00
parent 70a91ec3ba
commit 655cf17b60
19 changed files with 320 additions and 134 deletions
--- a/tests/cpp/data/test_sparse_page_dmatrix.cu
+++ b/tests/cpp/data/test_sparse_page_dmatrix.cu
@@ -86,7 +86,7 @@ struct ReadRowFunction {
      : matrix(std::move(matrix)), row(row), row_data_d(row_data_d) {}

  __device__ void operator()(size_t col) {
-    auto value = matrix.GetElement(row, col);
+    auto value = matrix.GetFvalue(row, col);
    if (isnan(value)) {
      value = -1;
    }