no need to restore stack pointer if we're exiting

diff --git a/hylo-lang/hyloc/src/riscv_asm_codegen.rs b/hylo-lang/hyloc/src/riscv_asm_codegen.rs
index d90ca6b0ca043f101ffc6633717e02f3330a52b1..3f0a96ca46776271eb775d5cece5ba39dc73e702 100644 (file)
--- a/hylo-lang/hyloc/src/riscv_asm_codegen.rs
+++ b/hylo-lang/hyloc/src/riscv_asm_codegen.rs
@@ -130,14 +130,14 @@ impl<'a> CodeGen<'a> {
                     self.line(format!("call {}", name));    
                 },
                 IR::Ret => {
-                    self.line("lw ra, 8(sp)");  // load return address from stack
-                    self.line("addi sp, sp, 16"); // restore stack pointer
                     if last_label == "main" {
                         // exit 0 syscall
                         self.line("li a7, 93");
                         self.line("mv a0, x0");
                         self.line("ecall");
                     } else {
+                        self.line("lw ra, 8(sp)");  // load return address from stack
+                        self.line("addi sp, sp, 16"); // restore stack pointer
                         self.line("ret");
                     }
                 },