#649: LBManager: Start rearranging to eliminate use of addAction

DARMA-tasking · Jul 28, 2020 · 0d46659 · 0d46659
1 parent c69d6b5
commit 0d46659
Showing 1 changed file with 2 additions and 2 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
@@ -149,6 +149,8 @@ LBManager::makeLB(MsgSharedPtr<StartLBMsg> msg) {
   auto base_proxy = proxy.template registerBaseCollective<lb::BaseLB>();
   auto phase = msg->getPhase();
 
+  destroy_lb_ = [proxy]{ proxy.destroyCollective(); };
+
   EpochType model_epoch = theTerm()->makeEpochCollective("LBManager::model_epoch");
   EpochType balance_epoch = theTerm()->makeEpochCollective("LBManager::balance_epoch");
   EpochType migrate_epoch = theTerm()->makeEpochCollective("LBManager::migrate_epoch");
@@ -189,8 +191,6 @@ LBManager::makeLB(MsgSharedPtr<StartLBMsg> msg) {
     this->finishedRunningLB(phase);
   });
 
-  destroy_lb_ = [proxy]{ proxy.destroyCollective(); };
-
   runSchedulerThrough(migrate_epoch);
 
   return proxy;