Merge pull request #1758 from DARMA-tasking/1757-allow-destroy-lb-out…

…side-manager 1757 Allow destroying an LB from outside the LBManager
DARMA-tasking · May 3, 2022 · 3030d38 · 3030d38
2 parents 918ca75 + 3381f68
commit 3030d38
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/src/vt/vrt/collection/balance/lb_invoke/lb_manager.cc b/src/vt/vrt/collection/balance/lb_invoke/lb_manager.cc
@@ -378,6 +378,14 @@ void LBManager::startup() {
   });
 }
 
+void LBManager::destroyLB() {
+  // Destruct the objgroup that was used for LB
+  if (destroy_lb_ != nullptr) {
+    destroy_lb_();
+    destroy_lb_ = nullptr;
+  }
+}
+
 void LBManager::finishedLB(PhaseType phase) {
   vt_debug_print(
     normal, lb,
@@ -387,11 +395,7 @@ void LBManager::finishedLB(PhaseType phase) {
   theNodeLBData()->startIterCleanup(phase, model_->getNumPastPhasesNeeded());
   theNodeLBData()->outputLBDataForPhase(phase);
 
-  // Destruct the objgroup that was used for LB
-  if (destroy_lb_ != nullptr) {
-    destroy_lb_();
-    destroy_lb_ = nullptr;
-  }
+  destroyLB();
 }
 
 void LBManager::statsHandler(StatsMsgType* msg) {

diff --git a/src/vt/vrt/collection/balance/lb_invoke/lb_manager.h b/src/vt/vrt/collection/balance/lb_invoke/lb_manager.h
@@ -166,6 +166,8 @@ struct LBManager : runtime::component::Component<LBManager> {
    */
   static void printLBArgsHelp(std::string lb);
 
+  void destroyLB();
+
 protected:
   /**
    * \internal