From 9701e605d34f0b96ed78ea8341e2a3b17ba7f1c1 Mon Sep 17 00:00:00 2001 From: fis Date: Tue, 1 Jun 2021 17:41:34 +0800 Subject: [PATCH] Use div roundup instead. --- src/predictor/cpu_predictor.cc | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/predictor/cpu_predictor.cc b/src/predictor/cpu_predictor.cc index 27d8f6c5dc63..d95bdf583a04 100644 --- 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(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;