#1129: scheduler: add subphase collectives

DARMA-tasking · May 13, 2022 · 7999707 · 7999707
1 parent d4f04ee
commit 7999707
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/vt/scheduler/scheduler.h b/src/vt/scheduler/scheduler.h
@@ -76,6 +76,12 @@ void runInEpochCollective(Callable&& fn);
 template <typename Callable>
 void runInEpochCollective(std::string const& label, Callable&& fn);
 
+template <typename Callable>
+void runSubphaseCollective(Callable&& fn);
+
+template <typename Callable>
+void runSubphaseCollective(std::string const& label, Callable&& fn);
+
 namespace messaging {
 
 template <typename T>

diff --git a/src/vt/scheduler/scheduler.impl.h b/src/vt/scheduler/scheduler.impl.h
@@ -47,6 +47,7 @@
 #include "vt/config.h"
 #include "vt/messaging/active.h"
 #include "vt/termination/termination.h"
+#include "vt/phase/phase_manager.h"
 
 namespace vt {
 
@@ -72,6 +73,20 @@ void runInEpochCollective(std::string const& label, Callable&& fn) {
   runInEpoch(ep, std::forward<Callable>(fn));
 }
 
+template <typename Callable>
+void runSubphaseCollective(Callable&& fn) {
+  runSubphaseCollective("UNLABELED", std::forward<Callable>(fn));
+}
+
+template <typename Callable>
+void runSubphaseCollective(std::string const& label, Callable&& fn) {
+  auto ep = theTerm()->makeEpochCollective(label);
+  theTerm()->addActionEpoch(ep, [=]{
+    thePhase()->advanceSubphase();
+  });
+  runInEpoch(ep, std::forward<Callable>(fn));
+}
+
 template <typename Callable>
 void runInEpochRooted(Callable&& fn) {
   runInEpochRooted("UNLABELED", std::forward<Callable>(fn));