#2281: Add template param for Rabenseifner to determine how to procee…

…d with handler
DARMA-tasking · Aug 25, 2024 · 05d8cb6 · 05d8cb6
1 parent 434a2b7
commit 05d8cb6
Show file tree

Hide file tree

Showing 7 changed files with 120 additions and 90 deletions.
diff --git a/src/vt/collective/reduce/allreduce/rabenseifner.h b/src/vt/collective/reduce/allreduce/rabenseifner.h
@@ -61,6 +61,10 @@
 
 namespace vt::collective::reduce::allreduce {
 
+struct CollectionAllreduceT {};
+struct GroupAllreduceT {};
+struct ObjgroupAllreduceT {};
+
 /**
  * \struct Rabenseifner
  * \brief Class implementing Rabenseifner's allreduce algorithm.
@@ -74,7 +78,7 @@ namespace vt::collective::reduce::allreduce {
  * \tparam finalHandler Callback handler for the final result.
  */
 template <
-  typename DataT, template <typename Arg> class Op, auto f
+  typename Type, typename DataT, template <typename Arg> class Op, auto f
 >
 struct Rabenseifner {
   using Data = DataT;