#941: added subphase support to loadComm

DARMA-tasking · Oct 15, 2020 · 67578f8 · 67578f8
1 parent 0a2631e
commit 67578f8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/vt/vrt/collection/balance/elm_stats.cc b/src/vt/vrt/collection/balance/elm_stats.cc
@@ -182,14 +182,14 @@ TimeType ElementStats::getLoad(PhaseType phase, SubphaseType subphase) const {
 }
 
 CommMapType const&
-ElementStats::getComm(PhaseType const& phase) {
+ElementStats::getComm(PhaseType const& phase, SubphaseType const& subphase) {
   comm_.resize(phase + 1);
   auto const& phase_comm = comm_[phase];
 
   vt_debug_print(
     lb, node,
-    "ElementStats: getComm: comm size={}, phase={}\n",
-    phase_comm.size(), phase
+    "ElementStats: getComm: comm size={}, phase={}, subphase={}\n",
+    phase_comm.size(), phase, subphase
   );
 
   return phase_comm;

diff --git a/src/vt/vrt/collection/balance/elm_stats.impl.h b/src/vt/vrt/collection/balance/elm_stats.impl.h
@@ -89,7 +89,7 @@ template <typename ColT>
   auto const& untyped_proxy = col->getProxy();
   auto const& total_load = stats.getLoad(cur_phase, getFocusedSubPhase(untyped_proxy));
   auto const& subphase_loads = stats.subphase_timings_.at(cur_phase);
-  auto const& comm = stats.getComm(cur_phase);
+  auto const& comm = stats.getComm(cur_phase, getFocusedSubPhase(untyped_proxy));
   auto const& idx = col->getIndex();
   auto const& elm_proxy = proxy[idx];