Use div roundup instead.

dmlc · Jun 1, 2021 · 9701e60 · 9701e60
1 parent dab1750
commit 9701e60
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/predictor/cpu_predictor.cc b/src/predictor/cpu_predictor.cc
@@ -187,9 +187,7 @@ void PredictBatchByBlockOfRowsKernel(
   // parallel over local batch
   const auto nsize = static_cast<bst_omp_uint>(batch.Size());
   const int num_feature = model.learner_model_param->num_feature;
-  const bst_omp_uint n_blocks =
-      (nsize) / block_of_rows_size + !!((nsize) % block_of_rows_size);
-  CHECK_EQ(common::DivRoundUp(nsize, block_of_rows_size), n_blocks);
+  omp_ulong n_blocks = common::DivRoundUp(nsize, block_of_rows_size);
 
   common::ParallelFor(n_blocks, [&](bst_omp_uint block_id) {
     const size_t batch_offset = block_id * block_of_rows_size;