#1507: Reset epoch_stack_size_ in TD::resume() in case underlying sta…

…ck has changed from when TD::suspend() was called
DARMA-tasking · Jul 22, 2021 · 5179013 · 5179013
1 parent e96ce0a
commit 5179013
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/vt/context/runnable_context/td.cc b/src/vt/context/runnable_context/td.cc
@@ -102,6 +102,9 @@ void TD::suspend() {
 void TD::resume() {
   theMsg()->pushEpoch(ep_);
 
+  auto& epoch_stack = theMsg()->getEpochStack();
+  base_epoch_stack_size_ = epoch_stack.size();
+
   for (auto it = suspended_epochs_.rbegin();
        it != suspended_epochs_.rend();
        ++it) {