fix: subi instead of pop in returnstmtNode

boozec · Jul 14, 2024 · ed58dc2 · ed58dc2
1 parent ce49d20
commit ed58dc2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ast/nodes/ReturnStmtNode.java b/src/ast/nodes/ReturnStmtNode.java
@@ -48,8 +48,8 @@ public Type typeCheck() {
     @Override
     public String codeGeneration() {
         String expS = exprList.codeGeneration();
-        for (int i = 0; i < localvar; i++) {
-            expS += "pop\n";
+        if(localvar > 0) {
+            expS += "subi SP " + localvar + "\n";
         }
         return expS +
                 "popr RA\n" +