Additions to the parallel algorithms.

Jhuighuy · Feb 16, 2025 · a57cb7c · a57cb7c
1 parent 919a5d4
commit a57cb7c
Show file tree

Hide file tree

Showing 11 changed files with 505 additions and 171 deletions.
diff --git a/source/tit/core/CMakeLists.txt b/source/tit/core/CMakeLists.txt
@@ -44,6 +44,7 @@ add_tit_library(
     "par/control.cpp"
     "par/control.hpp"
     "par/memory_pool.hpp"
+    "par/partitioner.hpp"
     "par/task_group.hpp"
     "profiler.cpp"
     "profiler.hpp"

diff --git a/source/tit/core/containers/multivector.hpp b/source/tit/core/containers/multivector.hpp
@@ -166,17 +166,20 @@ class Multivector {
   constexpr void assign_pairs_par_wide(size_t count, Pairs&& pairs) {
     TIT_ASSUME_UNIVERSAL(Pairs, pairs);
     assign_pairs_wide_impl_(count, [&pairs](auto func) {
-      par::static_for_each(pairs, std::move(func));
+      par::deterministic_for_each(pairs, std::move(func));
     });
   }
   template<par::range Range>
   constexpr void assign_pairs_par_wide(size_t count,
                                        std::ranges::join_view<Range> pairs) {
     auto base = std::move(pairs).base();
     assign_pairs_wide_impl_(count, [&base](auto func) {
-      par::static_for_each(base, [&func](size_t thread, const auto& range) {
-        std::ranges::for_each(range, std::bind_front(std::ref(func), thread));
-      });
+      par::deterministic_for_each(base,
+                                  [&func](const auto& range, size_t thread) {
+                                    std::ranges::for_each(
+                                        range,
+                                        std::bind_back(std::ref(func), thread));
+                                  });
     });
   }
   /// @}
@@ -223,7 +226,7 @@ class Multivector {
     static Mdvector<size_t, 2> per_thread_ranges{};
     const auto num_threads = par::num_threads();
     per_thread_ranges.assign(num_threads, count + 1);
-    for_each_pair([count](size_t thread, const auto& pair) {
+    for_each_pair([count](const auto& pair, size_t thread) {
       const auto index = std::get<0>(pair);
       TIT_ASSERT(index < count, "Index of the value is out of expected range!");
       per_thread_ranges[thread, index] += 1;
@@ -242,7 +245,7 @@ class Multivector {
     // Place each value into position of the first element of it's index
     // range, then increment the position.
     vals_.resize(val_ranges_.back());
-    for_each_pair([count, this](size_t thread, const auto& pair) {
+    for_each_pair([count, this](const auto& pair, size_t thread) {
       const auto& [index, value] = pair;
       TIT_ASSERT(index < count, "Index of the value is out of expected range!");
       auto& position = per_thread_ranges[thread, index];