From d411f98d2618cb1ee42a60b7e7e5f5e4be35d08d Mon Sep 17 00:00:00 2001 From: ShvetsKS <33296480+ShvetsKS@users.noreply.github.com> Date: Mon, 9 Nov 2020 12:01:30 +0300 Subject: [PATCH] simple fix for static shedule in predict (#6357) Co-authored-by: ShvetsKS --- src/predictor/cpu_predictor.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/predictor/cpu_predictor.cc b/src/predictor/cpu_predictor.cc index 7da841cc1..0999f2e08 100644 --- a/src/predictor/cpu_predictor.cc +++ b/src/predictor/cpu_predictor.cc @@ -154,7 +154,7 @@ void PredictBatchByBlockOfRowsKernel(DataView batch, std::vector *out const auto nsize = static_cast(batch.Size()); const bst_omp_uint n_row_blocks = (nsize) / block_of_rows_size + !!((nsize) % block_of_rows_size); -#pragma omp parallel for schedule(guided) +#pragma omp parallel for schedule(static) for (bst_omp_uint block_id = 0; block_id < n_row_blocks; ++block_id) { const size_t batch_offset = block_id * block_of_rows_size; const size_t block_size = std::min(nsize - batch_offset, block_of_rows_size);