M-Labs/nac3
M-Labs
/
nac3
8
0
Fork 10
Code Issues 99 Pull Requests 20 Releases Wiki Activity

WIP - [artiq] only change slot

This commit is contained in:
David Mak 2024-08-14 18:17:25 +08:00
parent 6afcd37b83
commit ad1b2dfd88
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nac3artiq/src/codegen.rs
View File

@ -680,7 +680,14 @@ fn format_rpc_ret<'ctx>(
// } // }
_ => { _ => {
let slot = ctx.builder.build_alloca(llvm_ret_ty, "rpc.ret.slot").unwrap(); let llvm_usize = generator.get_size_type(ctx.ctx);
let _slot = ctx.builder.build_alloca(llvm_ret_ty, "rpc.ret.slot").unwrap();
let slot = if matches!(&*ctx.unifier.get_ty_immutable(ret_ty), TypeEnum::TObj { obj_id, .. } if *obj_id == PrimDef::NDArray.id()) {
ctx.builder.build_array_alloca(llvm_i8, llvm_usize.const_int(8, false), "").unwrap()
} else {
_slot
};
let slotgen = ctx.builder.build_bitcast(slot, llvm_pi8, "rpc.ret.ptr").unwrap(); let slotgen = ctx.builder.build_bitcast(slot, llvm_pi8, "rpc.ret.ptr").unwrap();
ctx.builder.build_unconditional_branch(head_bb).unwrap(); ctx.builder.build_unconditional_branch(head_bb).unwrap();
ctx.builder.position_at_end(head_bb); ctx.builder.position_at_end(head_bb);
@ -706,7 +713,7 @@ fn format_rpc_ret<'ctx>(
ctx.builder.position_at_end(tail_bb); ctx.builder.position_at_end(tail_bb);
ctx.builder.build_load(slot, "rpc.result").unwrap() ctx.builder.build_load(_slot, "rpc.result").unwrap()
} }
}; };