#708: ProcStats: add method for LB strategies to get subphase loads a…

…s desired
DARMA-tasking · Jun 10, 2020 · 45a6f50 · 45a6f50
1 parent e508be0
commit 45a6f50
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/vt/vrt/collection/balance/proc_stats.cc b/src/vt/vrt/collection/balance/proc_stats.cc
@@ -109,6 +109,12 @@ ProcStats::getProcLoad(PhaseType phase) const {
   return proc_data_.at(phase);
 }
 
+ProcStats::SubphaseLoadMapType const&
+ProcStats::getProcSubphaseLoad(PhaseType phase) const {
+  vtAssert(proc_subphase_data_.size() > phase, "Phase must exist in load data");
+  return proc_subphase_data_.at(phase);
+}
+
 CommMapType const& ProcStats::getProcComm(PhaseType phase) const {
   vtAssert(proc_comm_.size() > phase, "Phase must exist in comm data");
   return proc_comm_.at(phase);

diff --git a/src/vt/vrt/collection/balance/proc_stats.h b/src/vt/vrt/collection/balance/proc_stats.h
@@ -136,6 +136,15 @@ struct ProcStats : runtime::component::Component<ProcStats> {
    */
   LoadMapType const& getProcLoad(PhaseType phase) const;
 
+  /**
+   * \internal \brief Get object loads for the subphases of a given phase
+   *
+   * \param[in] phase the phase
+   *
+   * \return the subphase load map
+   */
+  SubphaseLoadMapType const& getProcSubphaseLoad(PhaseType phase) const;
+
   /**
    * \internal \brief Get object comm graph for a given phase
    *