Merge pull request #203 from EpitechPromo2027/201-returning-void-is-b…

…roken fix: Function returning void
Jabolol · Jan 15, 2025 · e1133d9 · e1133d9
2 parents a93e3b1 + ecd33dc
commit e1133d9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/Codegen/Codegen.hs b/lib/Codegen/Codegen.hs
@@ -566,7 +566,9 @@ generateFunction (AT.Function _ name (AT.TFunction ret params var) paramNames bo
     oldAllocatedVars <- S.gets allocatedVars
     preAllocateVars body
     result <- generateExpr body
-    I.ret result
+    case ret of
+      AT.TVoid -> I.retVoid
+      _ -> I.ret result
     S.modify (\s -> s {allocatedVars = oldAllocatedVars})
   where
     mkParam t n = (toLLVM t, M.ParameterName $ U.stringToByteString n)