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

WIP - [artiq] Test

This commit is contained in:
David Mak 2024-08-15 04:59:43 +08:00
parent 7ba1560442
commit aaffe3cf9e
1 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
nac3artiq/src/codegen.rs
View File

@ -605,7 +605,7 @@ fn format_rpc_ret<'ctx>(
// let buffer_size = // let buffer_size =
// ctx.builder.build_int_add(dims_buf_sz, llvm_pdata_sizeof, "").unwrap(); // ctx.builder.build_int_add(dims_buf_sz, llvm_pdata_sizeof, "").unwrap();
let buffer_size = llvm_usize.const_int(32, false); let buffer_size = llvm_usize.const_int(8, false);
// let buffer_size = ctx.builder.build_left_shift( // let buffer_size = ctx.builder.build_left_shift(
// ctx.builder.build_int_add( // ctx.builder.build_int_add(
// ctx.builder.build_right_shift( // ctx.builder.build_right_shift(
@ -678,7 +678,7 @@ fn format_rpc_ret<'ctx>(
ctx, ctx,
generator, generator,
&llvm_usize.const_zero(), &llvm_usize.const_zero(),
llvm_usize.const_int(9, false), llvm_usize.const_int(6, false),
); );
} }
@ -693,24 +693,42 @@ fn format_rpc_ret<'ctx>(
// llvm_i1.const_zero(), // llvm_i1.const_zero(),
// ); // );
let sz = call_ndarray_calc_size(generator, ctx, &ndarray.dim_sizes(), (None, None));
ndarray.create_data( ndarray.create_data(
ctx, ctx,
llvm_elem_ty, llvm_elem_ty,
call_ndarray_calc_size(generator, ctx, &ndarray.dim_sizes(), (None, None)), sz,
);
let sz_addr = ctx.builder.build_alloca(llvm_usize, "").unwrap();
ctx.builder.build_store(sz_addr, sz).unwrap();
let ndarray_pdata = unsafe {
ndarray.data().base_ptr(ctx, generator)
};
call_memcpy_generic(
ctx,
ndarray_pdata,
sz_addr,
llvm_usize_sizeof,
llvm_i1.const_zero(),
); );
let alloc_size_addr = ctx.builder.build_alloca(llvm_usize, "").unwrap(); let alloc_size_addr = ctx.builder.build_alloca(llvm_usize, "").unwrap();
ctx.builder.build_store(alloc_size_addr, alloc_size).unwrap(); ctx.builder.build_store(alloc_size_addr, alloc_size).unwrap();
let ndarray_pdata = unsafe {
ndarray.data().ptr_offset_unchecked(ctx, generator, &llvm_usize.const_int(1, false), None)
};
call_memcpy_generic( call_memcpy_generic(
ctx, ctx,
ndarray.data().base_ptr(ctx, generator), ndarray_pdata,
alloc_size_addr, alloc_size_addr,
llvm_usize_sizeof, llvm_usize_sizeof,
llvm_i1.const_zero(), llvm_i1.const_zero(),
); );
let ndarray_pdata = unsafe { let ndarray_pdata = unsafe {
ndarray.data().ptr_offset_unchecked(ctx, generator, &llvm_usize.const_int(1, false), None) ndarray.data().ptr_offset_unchecked(ctx, generator, &llvm_usize.const_int(2, false), None)
}; };
call_memcpy_generic( call_memcpy_generic(
ctx, ctx,