fix device sketch with weights in external memory mode (#5870)

2020-07-07 17:44:07 -07:00
parent d0a29c3135
commit 06320729d4
2 changed files with 17 additions and 1 deletions
--- a/src/common/hist_util.cu
+++ b/src/common/hist_util.cu
@@ -264,7 +264,6 @@ void ProcessWeightedBatch(int device, const SparsePage& page,
        d_temp_weights[idx] = weights[group];
      });
  } else {
-    CHECK_EQ(weights.size(), page.offset.Size() - 1);
    dh::LaunchN(device, temp_weights.size(), [=] __device__(size_t idx) {
        size_t element_idx = idx + begin;
        size_t ridx = thrust::upper_bound(thrust::seq, row_ptrs.begin(),