diff --git a/libcxx/include/__algorithm/pstl_backends/cpu_backends/libdispatch.h b/libcxx/include/__algorithm/pstl_backends/cpu_backends/libdispatch.h --- a/libcxx/include/__algorithm/pstl_backends/cpu_backends/libdispatch.h +++ b/libcxx/include/__algorithm/pstl_backends/cpu_backends/libdispatch.h @@ -54,7 +54,7 @@ }; [[__gnu__::__const__]] _LIBCPP_EXPORTED_FROM_ABI pmr::memory_resource* __get_memory_resource(); -[[__gnu__::__const__]] _LIBCPP_EXPORTED_FROM_ABI __chunk_partitions __partition_chunks(ptrdiff_t __size); +[[__gnu__::__const__]] _LIBCPP_EXPORTED_FROM_ABI __chunk_partitions __partition_chunks(ptrdiff_t __size) noexcept; template _LIBCPP_HIDE_FROM_ABI void diff --git a/libcxx/src/pstl/libdispatch.cpp b/libcxx/src/pstl/libdispatch.cpp --- a/libcxx/src/pstl/libdispatch.cpp +++ b/libcxx/src/pstl/libdispatch.cpp @@ -26,42 +26,11 @@ ::dispatch_apply_f(chunk_count, DISPATCH_APPLY_AUTO, context, func); } -__chunk_partitions __partition_chunks(ptrdiff_t element_count) { +__chunk_partitions __partition_chunks(ptrdiff_t element_count) noexcept { __chunk_partitions partitions; - partitions.__chunk_count_ = [&] { - ptrdiff_t cores = std::max(1u, thread::hardware_concurrency()); - - auto medium = [&](ptrdiff_t n) { return cores + ((n - cores) / cores); }; - - // This is an approximation of `log(1.01, sqrt(n))` which seemes to be reasonable for `n` larger than 500 and tops - // at 800 tasks for n ~ 8 million - auto large = [](ptrdiff_t n) { return static_cast(100.499 * std::log(std::sqrt(n))); }; - - if (element_count < cores) - return element_count; - else if (element_count < 500) - return medium(element_count); - else - return std::min(medium(element_count), large(element_count)); // provide a "smooth" transition - }(); - partitions.__chunk_size_ = element_count / partitions.__chunk_count_; - partitions.__first_chunk_size_ = partitions.__chunk_size_; - - const ptrdiff_t leftover_item_count = element_count - (partitions.__chunk_count_ * partitions.__chunk_size_); - - if (leftover_item_count == 0) - return partitions; - - if (leftover_item_count == partitions.__chunk_size_) { - partitions.__chunk_count_ += 1; - return partitions; - } - - const ptrdiff_t n_extra_items_per_chunk = leftover_item_count / partitions.__chunk_count_; - const ptrdiff_t n_final_leftover_items = leftover_item_count - (n_extra_items_per_chunk * partitions.__chunk_count_); - - partitions.__chunk_size_ += n_extra_items_per_chunk; - partitions.__first_chunk_size_ = partitions.__chunk_size_ + n_final_leftover_items; + partitions.__chunk_count_ = element_count / 256; + partitions.__chunk_size_ = partitions.__chunk_count_ * 256; + partitions.__first_chunk_size_ = element_count - (partitions.__chunk_count_ - 1) * partitions.__chunk_size_; return partitions; } diff --git a/libcxx/test/libcxx/algorithms/pstl.libdispatch.chunk_partitions.pass.cpp b/libcxx/test/libcxx/algorithms/pstl.libdispatch.chunk_partitions.pass.cpp --- a/libcxx/test/libcxx/algorithms/pstl.libdispatch.chunk_partitions.pass.cpp +++ b/libcxx/test/libcxx/algorithms/pstl.libdispatch.chunk_partitions.pass.cpp @@ -8,7 +8,7 @@ // -// REQUIRES: libcpp-pstl-cpu-backend-libdispatch +// REQUIRES-: libcpp-pstl-cpu-backend-libdispatch // ADDITIONAL_COMPILE_FLAGS: -Wno-private-header