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 3b3af8cba9
1 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
nac3artiq/src/codegen.rs
View File

@ -693,24 +693,42 @@ fn format_rpc_ret<'ctx>(
// llvm_i1.const_zero(),
// );
let sz = call_ndarray_calc_size(generator, ctx, &ndarray.dim_sizes(), (None, None));
ndarray.create_data(
ctx,
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();
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(
ctx,
ndarray.data().base_ptr(ctx, generator),
ndarray_pdata,
alloc_size_addr,
llvm_usize_sizeof,
llvm_i1.const_zero(),
);
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(
ctx,