#1122: collection: expire unneeded stats

DARMA-tasking · Nov 18, 2020 · f49a2af · f49a2af
1 parent 92c5e4b
commit f49a2af
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/src/vt/vrt/collection/balance/elm_stats.cc b/src/vt/vrt/collection/balance/elm_stats.cc
@@ -202,6 +202,15 @@ SubphaseType ElementStats::getSubPhase() const {
   return cur_subphase_;
 }
 
+void ElementStats::startIterCleanup(PhaseType phase, unsigned int look_back) {
+  if (phase >= look_back) {
+    phase_timings_.erase(phase - look_back);
+    subphase_timings_.erase(phase - look_back);
+    comm_.erase(phase - look_back);
+    subphase_comm_.erase(phase - look_back);
+  }
+}
+
 std::size_t ElementStats::getLoadPhaseCount() const {
   return phase_timings_.size();
 }

diff --git a/src/vt/vrt/collection/balance/elm_stats.h b/src/vt/vrt/collection/balance/elm_stats.h
@@ -92,6 +92,11 @@ struct ElementStats {
   void setSubPhase(SubphaseType subphase);
   SubphaseType getSubPhase() const;
 
+  /**
+   * \internal \brief Cleanup after LB runs
+   */
+  void startIterCleanup(PhaseType phase, unsigned int look_back);
+
   // these are just for unit testing
   std::size_t getLoadPhaseCount() const;
   std::size_t getCommPhaseCount() const;

diff --git a/src/vt/vrt/collection/balance/elm_stats.impl.h b/src/vt/vrt/collection/balance/elm_stats.impl.h
@@ -95,6 +95,9 @@ void ElementStats::syncNextPhase(CollectStatsMsg<ColT>* msg, ColT* col) {
   theNodeStats()->addNodeStats(
     col, cur_phase, total_load, subphase_loads, comm, subphase_comm
   );
+
+  auto model = theLBManager()->getLoadModel();
+  stats.startIterCleanup(cur_phase, model->getNumPastPhasesNeeded());
 }
 
 }}}} /* end namespace vt::vrt::collection::balance */