Fix bug in flow optmization.

data61 · Nov 11, 2024 · 94b8c33 · 94b8c33
1 parent 3b166ca
commit 94b8c33
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Compiler/compilerLib.py b/Compiler/compilerLib.py
@@ -443,6 +443,8 @@ def compile_file(self):
                         continue
                     m = re.match(r"(\s*)if(\W.*):", line)
                     if m:
+                        while if_stack and if_stack[-1][0] == m.group(1):
+                            if_stack.pop()
                         if_stack.append((m.group(1), len(output)))
                         output.append("%s@if_(%s)\n" % (m.group(1), m.group(2)))
                         output.append("%sdef _():\n" % (m.group(1)))