factor out custom max_element

eth-cscs · Dec 2, 2024 · a50dcef · a50dcef
1 parent 2816d12
commit a50dcef
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/include/dlaf/auxiliary/norm/mc.h b/include/dlaf/auxiliary/norm/mc.h
@@ -25,6 +25,12 @@
 
 namespace dlaf::auxiliary::internal {
 
+template <class T>
+T max_element(std::vector<T>&& values) {
+  DLAF_ASSERT(!values.empty(), "");
+  return *std::max_element(values.begin(), values.end());
+}
+
 template <class T>
 pika::execution::experimental::unique_any_sender<T> reduce_in_place(
     pika::execution::experimental::unique_any_sender<dlaf::comm::CommunicatorPipelineExclusiveWrapper>
@@ -102,17 +108,12 @@ pika::execution::experimental::unique_any_sender<dlaf::BaseType<T>> Norm<
 
   // then it is necessary to reduce max values from all ranks into a single max value for the matrix
 
-  auto max_element = [](std::vector<NormT>&& values) {
-    DLAF_ASSERT(!values.empty(), "");
-    return *std::max_element(values.begin(), values.end());
-  };
-
   ex::unique_any_sender<NormT> local_max_value = ex::just(NormT{0});
   if (!tiles_max.empty())
     local_max_value =
         ex::when_all_vector(std::move(tiles_max)) |
         dlaf::internal::transform(dlaf::internal::Policy<Backend::MC>(thread_stacksize::nostack),
-                                  std::move(max_element));
+                                  max_element<NormT>);
 
   return reduce_in_place(comm_grid.full_communicator_pipeline().exclusive(),
                          comm_grid.rankFullCommunicator(rank), MPI_MAX, std::move(local_max_value));
@@ -149,16 +150,12 @@ pika::execution::experimental::unique_any_sender<dlaf::BaseType<T>> Norm<
 
   // then it is necessary to reduce max values from all ranks into a single max value for the matrix
 
-  auto max_element = [](std::vector<NormT>&& values) {
-    DLAF_ASSERT(!values.empty(), "");
-    return *std::max_element(values.begin(), values.end());
-  };
   ex::unique_any_sender<NormT> local_max_value = ex::just(NormT{0});
   if (!tiles_max.empty())
     local_max_value =
         ex::when_all_vector(std::move(tiles_max)) |
         dlaf::internal::transform(dlaf::internal::Policy<Backend::MC>(thread_stacksize::nostack),
-                                  std::move(max_element));
+                                  max_element<NormT>);
 
   return reduce_in_place(comm_grid.full_communicator_pipeline().exclusive(),
                          comm_grid.rankFullCommunicator(rank), MPI_MAX, std::move(local_max_value));