From 668d7338aec029bf789ce8c0e299943b8299aa5c Mon Sep 17 00:00:00 2001
From: Guy C <gc1523@ic.ac.uk>
Date: Mon, 24 Feb 2025 19:47:06 +0000
Subject: [PATCH] feat: move default return out of functionEpilogue into main
 def

---
 src/main/wacc/backend/asmGenerator.scala | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/wacc/backend/asmGenerator.scala b/src/main/wacc/backend/asmGenerator.scala
index 16b9a41..4c1a924 100644
--- a/src/main/wacc/backend/asmGenerator.scala
+++ b/src/main/wacc/backend/asmGenerator.scala
@@ -25,6 +25,7 @@ object asmGenerator {
         funcPrologue() ++
         alignStack() ++
         main.flatMap(generateStmt) ++
+        List(Move(Register(RegSize.R64, RegName.AX), ImmediateVal(0))) ++
         funcEpilogue() ++
         List(assemblyIR.Return()) ++
         generateFuncs()
@@ -204,7 +205,6 @@ object asmGenerator {
 
   def funcEpilogue(): List[AsmLine] = {
     List(
-      Move(Register(RegSize.R64, RegName.AX), ImmediateVal(0)),
       Move(Register(RegSize.R64, RegName.SP), Register(RegSize.R64, RegName.BP)),
       Pop(Register(RegSize.R64, RegName.BP))
     )