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

* Use explicit template parameter for specifying page type.
This commit is contained in:
Rong Ou
2019-08-03 12:10:34 -07:00
committed by Jiaming Yuan
parent e930a8e54f
commit 6edddd7966
41 changed files with 477 additions and 470 deletions

View File

@@ -440,7 +440,7 @@ struct GPUSketcher {
row_stride_ = 0;
sketch_container_.reset(new SketchContainer(param_, dmat));
for (const auto &batch : dmat->GetRowBatches()) {
for (const auto &batch : dmat->GetBatches<SparsePage>()) {
this->SketchBatch(batch, info);
}