[Wasm Exceptions] Handle delegation to the caller in RemoveUnusedNames (#3585)

1 files changed, 6 insertions, 1 deletions

diff --git a/src/passes/RemoveUnusedNames.cpp b/src/passes/RemoveUnusedNames.cpp
index addd4367b..a08cb5eda 100644
--- a/src/passes/RemoveUnusedNames.cpp
+++ b/src/passes/RemoveUnusedNames.cpp
@@ -21,6 +21,7 @@
 
 #include <ir/branch-utils.h>
 #include <pass.h>
+#include <shared-constants.h>
 #include <wasm.h>
 
 namespace wasm {
@@ -85,7 +86,11 @@ struct RemoveUnusedNames
     visitExpression(curr);
   }
 
-  void visitFunction(Function* curr) { assert(branchesSeen.empty()); }
+  void visitFunction(Function* curr) {
+    // When we reach the function body we can erase delegations to the caller.
+    branchesSeen.erase(DELEGATE_CALLER_TARGET);
+    assert(branchesSeen.empty());
+  }
 };
 
 Pass* createRemoveUnusedNamesPass() { return new RemoveUnusedNames(); }