Merge pull request #127 from certik/exit2

certik · web-flow · commit f4cf11ef5021 · 2022-02-10T12:16:05.000-07:00
LLVM: take into account the exit code in Stop
diff --git a/src/libasr/codegen/asr_to_llvm.cpp b/src/libasr/codegen/asr_to_llvm.cpp
@@ -3472,12 +3472,11 @@ class ASRToLLVMVisitor : public ASR::BaseVisitor<ASRToLLVMVisitor>
         printf(context, *module, *builder, printf_args);
     }
 
-    void visit_Stop(const ASR::Stop_t & /* x */) {
+    void visit_Stop(const ASR::Stop_t &x) {
         llvm::Value *fmt_ptr = builder->CreateGlobalStringPtr("STOP\n");
         printf(context, *module, *builder, {fmt_ptr});
-        int exit_code_int = 0;
-        llvm::Value *exit_code = llvm::ConstantInt::get(context,
-                llvm::APInt(32, exit_code_int));
+        this->visit_expr(*x.m_code);
+        llvm::Value *exit_code = tmp;
         exit(context, *module, *builder, exit_code);
     }
 

Original file line number	Diff line number	Diff line change
`@@ -3472,12 +3472,11 @@ class ASRToLLVMVisitor : public ASR::BaseVisitor<ASRToLLVMVisitor>`
`3472`	`3472`	`printf(context, module, builder, printf_args);`
`3473`	`3473`	`}`
`3474`	`3474`
`3475`		`- void visit_Stop(const ASR::Stop_t & /* x */) {`
	`3475`	`+ void visit_Stop(const ASR::Stop_t &x) {`
`3476`	`3476`	`llvm::Value *fmt_ptr = builder->CreateGlobalStringPtr("STOP\n");`
`3477`	`3477`	`printf(context, module, builder, {fmt_ptr});`
`3478`		`- int exit_code_int = 0;`
`3479`		`- llvm::Value *exit_code = llvm::ConstantInt::get(context,`
`3480`		`- llvm::APInt(32, exit_code_int));`
	`3478`	`+ this->visit_expr(*x.m_code);`
	`3479`	`+ llvm::Value *exit_code = tmp;`
`3481`	`3480`	`exit(context, module, builder, exit_code);`
`3482`	`3481`	`}`
`3483`	`3482`