From 9d78caf6d918489b04bb9add2a14254a876e2d51 Mon Sep 17 00:00:00 2001
From: Guy C <gc1523@ic.ac.uk>
Date: Mon, 24 Feb 2025 18:57:13 +0000
Subject: [PATCH] feat: add support for return statements in asmGenerator

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

diff --git a/src/main/wacc/backend/asmGenerator.scala b/src/main/wacc/backend/asmGenerator.scala
index c8981af..16b9a41 100644
--- a/src/main/wacc/backend/asmGenerator.scala
+++ b/src/main/wacc/backend/asmGenerator.scala
@@ -117,6 +117,8 @@ object asmGenerator {
           body.flatMap(generateStmt) ++
           List(Jump(LabelArg(startLabel)), LabelDef(endLabel))
       }
+      case microWacc.Return(expr) =>
+        evalExprIntoReg(expr, Register(RegSize.R64, RegName.AX))
       case _ => List()
     }