Refactor DMatrix to return batches of different page types (#4686)

* Use explicit template parameter for specifying page type.
2019-08-03 12:10:34 -07:00
parent e930a8e54f
commit 6edddd7966
41 changed files with 477 additions and 470 deletions
--- a/src/tree/updater_gpu_hist.cu
+++ b/src/tree/updater_gpu_hist.cu
@@ -1382,7 +1382,7 @@ class GPUHistMakerSpecialised {

    monitor_.StartCuda("BinningCompression");
    DeviceHistogramBuilderState hist_builder_row_state(shards_);
-    for (const auto &batch : dmat->GetRowBatches()) {
+    for (const auto &batch : dmat->GetBatches<SparsePage>()) {
      hist_builder_row_state.BeginBatch(batch);

      dh::ExecuteIndexShards(