Function ParallelForBlock

src/common/threading_utils.h:275–286

/** @brief Use n_threads as the number of blocks. */
template <typename Index, typename Func>
void ParallelForBlock(Index size, std::int32_t n_threads, Func&& fn) {
  static_assert(std::is_void_v<std::invoke_result_t<Func, common::Range1d>>);
  std::size_t blk_size = size / n_threads + (size % n_threads > 0);
  ParallelFor(n_threads, n_threads, [&](auto tid) {
    auto blk_beg = tid * blk_size;
    auto blk_end = std::min((tid + 1) * blk_size, static_cast<std::size_t>(size));
    if (blk_end <= blk_beg) {
      return;
    }
    fn(common::Range1d{blk_beg, blk_end});
  });
}

inline std::int32_t OmpGetThreadLimit() {
  std::int32_t limit = omp_get_thread_limit();