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

WIP - [artiq] More test

This commit is contained in:
David Mak 2024-08-15 04:03:06 +08:00
parent 758ea48a4b
commit 4522077644
1 changed files with 3 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
nac3artiq/src/codegen.rs
View File

@ -648,7 +648,7 @@ fn format_rpc_ret<'ctx>(
.build_load(i_addr, "")
.map(BasicValueEnum::into_int_value)
.unwrap();
Ok(ctx
.builder
.build_int_compare(IntPredicate::EQ, i, llvm_usize.const_zero(), "")
@ -667,7 +667,7 @@ fn format_rpc_ret<'ctx>(
llvm_pdata_sizeof,
llvm_i1.const_zero(),
);
let pbuffer_dims_begin = unsafe {
buffer.ptr_offset_unchecked(ctx, generator, &llvm_pdata_sizeof, None)
};
@ -678,31 +678,7 @@ fn format_rpc_ret<'ctx>(
dims_buf_sz,
llvm_i1.const_zero(),
);
// call_memcpy_generic(
// ctx,
// ndarray.dim_sizes().base_ptr(ctx, generator),
// buffer.base_ptr(ctx, generator),
// dims_buf_sz,
// llvm_i1.const_zero(),
// );
//
// let ppdata =
// generator.gen_var_alloc(ctx, llvm_ret_ty.element_type(), None).unwrap();
// let pbuffer_data_begin = unsafe {
// buffer.ptr_offset_unchecked(ctx, generator, &dims_buf_sz, None)
// };
// ctx.builder
// .build_store(ppdata, ndarray.data().base_ptr(ctx, generator))
// .unwrap();
// call_memcpy_generic(
// ctx,
// ppdata,
// pbuffer_data_begin,
// llvm_pdata_sizeof,
// llvm_i1.const_zero(),
// );
Ok(())
},
|_, _| Ok(()),
@ -717,13 +693,6 @@ fn format_rpc_ret<'ctx>(
ctx.builder.build_conditional_branch(is_done, tail_bb, alloc_bb).unwrap();
ctx.builder.position_at_end(alloc_bb);
// let alloc_size = ctx.builder.build_right_shift(
// ctx.builder.build_left_shift(
//
// ).unwrap(),
// llvm_usize.const_int(2, false),
// "",
// ).unwrap();
let alloc_size = ctx.builder.build_left_shift(
ctx.builder.build_int_add(
ctx.builder.build_right_shift(
@ -740,8 +709,6 @@ fn format_rpc_ret<'ctx>(
).unwrap();
let alloc_ptr =
ctx.builder.build_array_alloca(llvm_i8, alloc_size, "rpc.alloc").unwrap();
// let alloc_ptr =
// ctx.builder.build_bitcast(alloc_ptr, llvm_pi8, "rpc.alloc.ptr").unwrap();
phi.add_incoming(&[(&alloc_ptr, alloc_bb)]);
let i =
ctx.builder.build_load(i_addr, "i").map(BasicValueEnum::into_int_value).unwrap();