#276: reduce: reorder arguments for bare reduce to match

DARMA-tasking · Jun 13, 2023 · e454bc5 · e454bc5
1 parent 1aea01b
commit e454bc5
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 19 deletions.
diff --git a/examples/hello_world/hello_reduce.cc b/examples/hello_world/hello_reduce.cc
@@ -55,7 +55,7 @@ int main(int argc, char** argv) {
   vt::NodeType const root = 0;
 
   auto r = vt::theCollective()->global();
-  r->reduce<vt::collective::PlusOp, reduceResult>(vt::Node{root}, 50, 52.334);
+  r->reduce<reduceResult, vt::collective::PlusOp>(vt::Node{root}, 50, 52.334);
 
   vt::finalize();
   return 0;

diff --git a/src/vt/collective/reduce/get_reduce_stamp.h b/src/vt/collective/reduce/get_reduce_stamp.h
@@ -48,6 +48,7 @@
 #include "vt/collective/reduce/reduce_scope.h"
 #include "vt/messaging/message.h"
 
+#include <tuple>
 #include <utility>
 #include <type_traits>
 

diff --git a/src/vt/collective/reduce/reduce.h b/src/vt/collective/reduce/reduce.h
@@ -168,31 +168,27 @@ struct Reduce : virtual collective::tree::Tree {
     return reduce<MsgT, f>(root, msg, id, num_contrib);
   }
 
-  //////////////////////////////////////////////////////////////
-  //////////////////////////////////////////////////////////////
-  //////////////////////////////////////////////////////////////
-  template <template<typename Arg> class Op, auto f, typename... Params>
+  /**
+   * \brief Perform a reduction without a message
+   *
+   * \param[in] root the root node to target
+   * \param[in] params parameters to reduce
+   *
+   * \return a pending send
+   */
+  template <
+    auto f,
+    template <typename Arg> class Op = NoneOp,
+    typename... Params
+  >
   PendingSendType reduce(Node root, Params&&... params) {
     using Tuple = typename FuncTraits<decltype(f)>::TupleType;
     using OpT = Op<Tuple>;
     return reduce<OpT, f>(root, std::forward<Params>(params)...);
   }
 
   template <typename Op, auto f, typename... Params>
-  PendingSendType reduce(Node root, Params&&... params) {
-    using Tuple = typename FuncTraits<decltype(f)>::TupleType;
-    using MsgT = ReduceTMsg<Tuple>;
-
-    auto msg = vt::makeMessage<MsgT>(std::tuple{std::forward<Params>(params)...});
-    auto id = detail::ReduceStamp{};
-    auto han = auto_registry::makeAutoHandlerParam<decltype(f), f, MsgT>();
-    msg->root_handler_ = han;
-
-    return reduce<Op, operators::NoCombine, MsgT>(root.get(), msg.get(), id, 1);
-  }
-  //////////////////////////////////////////////////////////////
-  //////////////////////////////////////////////////////////////
-  //////////////////////////////////////////////////////////////
+  PendingSendType reduce(Node root, Params&&... params);
 
   /**
    * \brief Reduce a message up the tree

diff --git a/src/vt/collective/reduce/reduce.impl.h b/src/vt/collective/reduce/reduce.impl.h
@@ -83,6 +83,20 @@ void Reduce::reduceRootRecv(MsgT* msg) {
     .run();
 }
 
+template <typename Op, auto f, typename... Params>
+Reduce::PendingSendType Reduce::reduce(Node root, Params&&... params) {
+  using Tuple = typename FuncTraits<decltype(f)>::TupleType;
+  using MsgT = ReduceTMsg<Tuple>;
+
+  auto msg = vt::makeMessage<MsgT>(std::forward<Params>(params)...);
+  auto id = detail::ReduceStamp{};
+  auto han = auto_registry::makeAutoHandlerParam<decltype(f), f, MsgT>();
+  msg->root_handler_ = han;
+
+  return reduce<Op, operators::NoCombine, MsgT>(root.get(), msg.get(), id, 1);
+}
+
+
 template <typename OpT, typename MsgT, ActiveTypedFnType<MsgT> *f>
 Reduce::PendingSendType Reduce::reduce(
   NodeType const& root, MsgT* msg, Callback<MsgT> cb, detail::ReduceStamp id,