Unify CPU hist sketching (#5880)

2020-08-12 01:33:06 +08:00
parent bd6b7f4aa7
commit ee70a2380b
18 changed files with 648 additions and 677 deletions
--- a/src/common/hist_util.cuh
+++ b/src/common/hist_util.cuh
@@ -306,7 +306,7 @@ void AdapterDeviceSketch(Batch batch, int num_bins,
      size_t end = std::min(batch.Size(), size_t(begin + sketch_batch_num_elements));
      ProcessWeightedSlidingWindow(batch, info,
                                   num_cuts_per_feature,
-                                   CutsBuilder::UseGroup(info), missing, device, num_cols, begin, end,
+                                   HostSketchContainer::UseGroup(info), missing, device, num_cols, begin, end,
                                   sketch_container);
    }
  } else {