Handle duplicated values in sketching. (#6178)

* Accumulate weights in duplicated values. * Fix device id in iterative dmatrix.
2020-10-10 19:32:44 +08:00
parent ab5b35134f
commit 2241563f23
9 changed files with 250 additions and 54 deletions
--- a/src/common/hist_util.cu
+++ b/src/common/hist_util.cu
@@ -269,7 +269,7 @@ void ProcessWeightedBatch(int device, const SparsePage& page,
                             &cuts_ptr, &column_sizes_scan);

  auto const& h_cuts_ptr = cuts_ptr.ConstHostVector();
-  auto d_cuts_ptr = cuts_ptr.ConstDeviceSpan();
+  auto d_cuts_ptr = cuts_ptr.DeviceSpan();

  // Extract cuts
  sketch_container->Push(dh::ToSpan(sorted_entries),