#1183: reset the phase when restoring from file

DARMA-tasking · Dec 16, 2020 · 2340c2a · 2340c2a
1 parent 8acd899
commit 2340c2a
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/src/vt/vrt/collection/balance/elm_stats.cc b/src/vt/vrt/collection/balance/elm_stats.cc
@@ -141,6 +141,10 @@ void ElementStats::updatePhase(PhaseType const& inc) {
   subphase_comm_[cur_phase_];
 }
 
+void ElementStats::resetPhase() {
+  cur_phase_ = fst_lb_phase;
+}
+
 PhaseType ElementStats::getPhase() const {
   return cur_phase_;
 }

diff --git a/src/vt/vrt/collection/balance/elm_stats.h b/src/vt/vrt/collection/balance/elm_stats.h
@@ -83,6 +83,7 @@ struct ElementStats {
     double bytes, bool bcast
   );
   void updatePhase(PhaseType const& inc = 1);
+  void resetPhase();
   PhaseType getPhase() const;
   TimeType getLoad(PhaseType const& phase) const;
   TimeType getLoad(PhaseType phase, SubphaseType subphase) const;

diff --git a/src/vt/vrt/collection/manager.h b/src/vt/vrt/collection/manager.h
@@ -1836,6 +1836,8 @@ struct CollectionManager
   /**
    * \brief Restore the collection (collective) from file.
    *
+   * \note Resets the phase to 0 for every element.
+   *
    * \param[in] range the range of the collection to restart
    * \param[in] file_base the base file name for the files to read
    *

diff --git a/src/vt/vrt/collection/manager.impl.h b/src/vt/vrt/collection/manager.impl.h
@@ -3223,6 +3223,7 @@ CollectionManager::restoreFromFile(
     // @todo: error check the file read with bytes in directory
 
     auto col_ptr = checkpoint::deserializeFromFile<ColT>(file_name);
+    col_ptr->stats_.resetPhase();
     token[idx].insertPtr(std::move(col_ptr));
   }