fix: Fixed defer not working at top level

alinalihassan · May 22, 2022 · a5da7b2 · a5da7b2
1 parent c5fe7b6
commit a5da7b2
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/liblesma/Backend/Codegen.cpp b/src/liblesma/Backend/Codegen.cpp
@@ -219,8 +219,15 @@ int Codegen::JIT() {
 }
 
 void Codegen::Run() {
+    deferStack.push({});
     visit(Parser_->getAST());
 
+    auto instrs = deferStack.top();
+    deferStack.pop();
+
+    for (auto inst: instrs)
+        visit(inst);
+
     // Return 0 for top-level function
     Builder->CreateRet(ConstantInt::getSigned(Builder->getInt64Ty(), 0));
 }