Fix memory usage of device sketching (#5407)

2020-03-14 13:43:24 +13:00
parent bb8c8df39d
commit b745b7acce
13 changed files with 153 additions and 73 deletions
--- a/tests/cpp/tree/gpu_hist/test_histogram.cu
+++ b/tests/cpp/tree/gpu_hist/test_histogram.cu
@@ -13,7 +13,7 @@ void TestDeterminsticHistogram() {

  auto pp_m = CreateDMatrix(kRows, kCols, 0.5);
  auto& matrix = **pp_m;
-  BatchParam batch_param{0, static_cast<int32_t>(kBins), 0, 0};
+  BatchParam batch_param{0, static_cast<int32_t>(kBins), 0};

  for (auto const& batch : matrix.GetBatches<EllpackPage>(batch_param)) {
    auto* page = batch.Impl();