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

WIP - Constants

This commit is contained in:
David Mak 2024-08-13 17:51:26 +08:00
parent 07faed6779
commit 7d85b9a64b
1 changed files with 34 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
nac3artiq/src/codegen.rs
View File

@ -431,7 +431,8 @@ fn gen_rpc_tag(
); );
buffer.push(b'a'); buffer.push(b'a');
buffer.push((ndarray_ndims & 0xFF) as u8); buffer.push(1u8);
// buffer.push((ndarray_ndims & 0xFF) as u8);
gen_rpc_tag(ctx, ndarray_dtype, buffer)?; gen_rpc_tag(ctx, ndarray_dtype, buffer)?;
} }
_ => return Err(format!("Unsupported type: {:?}", ctx.unifier.stringify(ty))), _ => return Err(format!("Unsupported type: {:?}", ctx.unifier.stringify(ty))),
@ -581,23 +582,39 @@ fn rpc_codegen_callback_fn<'ctx>(
let buffer = ctx.builder.build_array_alloca(int8, buffer_size, "").unwrap(); let buffer = ctx.builder.build_array_alloca(int8, buffer_size, "").unwrap();
let buffer = ArraySliceValue::from_ptr_val(buffer, buffer_size, None); let buffer = ArraySliceValue::from_ptr_val(buffer, buffer_size, None);
llvm_intrinsics::call_memcpy_generic( for (offset, byte) in [128u8, 0u8, 0u8, 128u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8]
ctx, .into_iter()
buffer.base_ptr(ctx, generator), .enumerate()
llvm_arg.dim_sizes().base_ptr(ctx, generator), {
dims_buf_sz, let pbuffer = unsafe {
int1.const_zero(), buffer.ptr_offset_unchecked(
); ctx,
generator,
&size_type.const_int(offset as u64, false),
None,
)
};
let pbuffer_data_begin = ctx.builder.build_store(pbuffer, int8.const_int(byte as u64, false)).unwrap();
unsafe { buffer.ptr_offset_unchecked(ctx, generator, &dims_buf_sz, None) }; }
llvm_intrinsics::call_memcpy_generic(
ctx, // llvm_intrinsics::call_memcpy_generic(
pbuffer_data_begin, // ctx,
llvm_arg.data().base_ptr(ctx, generator), // buffer.base_ptr(ctx, generator),
data_buf_sz, // llvm_arg.dim_sizes().base_ptr(ctx, generator),
int1.const_zero(), // dims_buf_sz,
); // int1.const_zero(),
// );
//
// let pbuffer_data_begin =
// unsafe { buffer.ptr_offset_unchecked(ctx, generator, &dims_buf_sz, None) };
// llvm_intrinsics::call_memcpy_generic(
// ctx,
// pbuffer_data_begin,
// llvm_arg.data().base_ptr(ctx, generator),
// data_buf_sz,
// int1.const_zero(),
// );
buffer.base_ptr(ctx, generator) buffer.base_ptr(ctx, generator)
} else { } else {