From adb5c69e67bc6f9295a198c0a4661fdb45434fc3 Mon Sep 17 00:00:00 2001 From: ychenfo Date: Tue, 8 Mar 2022 21:50:28 +0800 Subject: [PATCH] nac3core: min builtin function --- nac3core/src/toplevel/builtins.rs | 67 ++++++++++++++++++++++++++++++- 1 file changed, 65 insertions(+), 2 deletions(-) diff --git a/nac3core/src/toplevel/builtins.rs b/nac3core/src/toplevel/builtins.rs index f773daf8..38cdd1c3 100644 --- a/nac3core/src/toplevel/builtins.rs +++ b/nac3core/src/toplevel/builtins.rs @@ -5,7 +5,7 @@ use crate::{ }, symbol_resolver::SymbolValue, }; -use inkwell::{FloatPredicate, IntPredicate}; +use inkwell::{types::BasicType, FloatPredicate, IntPredicate}; type BuiltinInfo = (Vec<(Arc>, Option)>, &'static [&'static str]); @@ -648,7 +648,7 @@ pub fn get_builtins(primitives: &mut (PrimitiveStore, Unifier)) -> BuiltinInfo { signature: primitives.1.add_ty(TypeEnum::TFunc(FunSignature { args: vec![FuncArg { name: "n".into(), ty: num_ty.0, default_value: None }], ret: primitives.0.bool, - vars: var_map, + vars: var_map.clone(), })), var_id: Default::default(), instance_to_symbol: Default::default(), @@ -916,6 +916,68 @@ pub fn get_builtins(primitives: &mut (PrimitiveStore, Unifier)) -> BuiltinInfo { loc: None, } })), + Arc::new(RwLock::new(TopLevelDef::Function { + name: "min".into(), + simple_name: "min".into(), + signature: primitives.1.add_ty(TypeEnum::TFunc(FunSignature { + args: vec![ + FuncArg { name: "m".into(), ty: num_ty.0, default_value: None }, + FuncArg { name: "n".into(), ty: num_ty.0, default_value: None }, + ], + ret: num_ty.0, + vars: var_map, + })), + var_id: Default::default(), + instance_to_symbol: Default::default(), + instance_to_stmt: Default::default(), + resolver: None, + codegen_callback: Some(Arc::new(GenCall::new(Box::new( + |ctx, _, fun, args, generator| { + let boolean = ctx.primitives.bool; + let int32 = ctx.primitives.int32; + let int64 = ctx.primitives.int64; + let uint32 = ctx.primitives.uint32; + let uint64 = ctx.primitives.uint64; + let float = ctx.primitives.float; + let llvm_i1 = ctx.ctx.bool_type().as_basic_type_enum(); + let llvm_i32 = ctx.ctx.i32_type().as_basic_type_enum(); + let llvm_i64 = ctx.ctx.i64_type().as_basic_type_enum(); + let llvm_f64 = ctx.ctx.f64_type().as_basic_type_enum(); + let m_ty = fun.0.args[0].ty; + let n_ty = fun.0.args[1].ty; + let m_val = args[0].1.clone().to_basic_value_enum(ctx, generator)?; + let n_val = args[1].1.clone().to_basic_value_enum(ctx, generator)?; + let mut is_type = |a: Type, b: Type| ctx.unifier.unioned(a, b); + let (fun_name, arg_ty) = if is_type(m_ty, n_ty) && is_type(n_ty, boolean) { + ("llvm.umin.i1", llvm_i1) + } else if is_type(m_ty, n_ty) && is_type(n_ty, int32) { + ("llvm.smin.i32", llvm_i32) + } else if is_type(m_ty, n_ty) && is_type(n_ty, int64) { + ("llvm.smin.i64", llvm_i64) + } else if is_type(m_ty, n_ty) && is_type(n_ty, uint32) { + ("llvm.umin.i32", llvm_i32) + } else if is_type(m_ty, n_ty) && is_type(n_ty, uint64) { + ("llvm.umin.i64", llvm_i64) + } else if is_type(m_ty, n_ty) && is_type(n_ty, float) { + ("llvm.minnum.f64", llvm_f64) + } else { + unreachable!(); + }; + let intrinsic = ctx.module.get_function(fun_name).unwrap_or_else(|| { + let fn_type = arg_ty.fn_type(&[arg_ty.into(), arg_ty.into()], false); + ctx.module.add_function(fun_name, fn_type, None) + }); + let val = ctx + .builder + .build_call(intrinsic, &[m_val.into(), n_val.into()], "min") + .try_as_basic_value() + .left() + .unwrap(); + Ok(val.into()) + }, + )))), + loc: None, + })), ]; let ast_list: Vec>> = (0..top_level_def_list.len()).map(|_| None).collect(); @@ -939,6 +1001,7 @@ pub fn get_builtins(primitives: &mut (PrimitiveStore, Unifier)) -> BuiltinInfo { "ceil", "ceil64", "len", + "min", ], ) }