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

60
nac3artiq/src/codegen.rs
View File

@ -32,7 +32,6 @@ use crate::{symbol_resolver::InnerResolver, timeline::TimeFns};
use inkwell::types::BasicType;
use itertools::Itertools;
use nac3core::codegen::classes::{ArrayLikeIndexer, ArrayLikeValue, ArraySliceValue, NDArrayType};
use nac3core::codegen::llvm_intrinsics;
use std::{
collections::{hash_map::DefaultHasher, HashMap},
hash::{Hash, Hasher},
@ -431,7 +430,8 @@ fn gen_rpc_tag(
);
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)?;
}
_ => return Err(format!("Unsupported type: {:?}", ctx.unifier.stringify(ty))),
@ -578,26 +578,50 @@ fn rpc_codegen_callback_fn<'ctx>(
let buffer_size = ctx.builder.build_int_add(dims_buf_sz, data_buf_sz, "").unwrap();
let buffer = ctx.builder.build_array_alloca(int8, buffer_size, "").unwrap();
// let buffer = ctx.builder.build_array_alloca(int8, buffer_size, "").unwrap();
let buffer = ctx.builder.build_array_alloca(int8, size_type.const_int(44, false), "").unwrap();
let buffer = ArraySliceValue::from_ptr_val(buffer, buffer_size, None);
llvm_intrinsics::call_memcpy_generic(
for (offset, byte) in [
/* LSB <-> MSB */
/* buffer[0] - ndims */ 2u8, 0u8, 0u8, 0u8,
/* buffer[4] - dim[0] */ 2u8, 0u8, 0u8, 0u8,
/* buffer[8] - dim[1] */ 2u8, 0u8, 0u8, 0u8,
/* buffer[12] - data[0] */ 128u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8,
/* buffer[20] - data[1] */ 0u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8,
/* buffer[28] - data[2] */ 128u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8,
/* buffer[36] - data[3] */ 0u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8,
].into_iter().enumerate()
{
let pbuffer = unsafe {
buffer.ptr_offset_unchecked(
ctx,
buffer.base_ptr(ctx, generator),
llvm_arg.dim_sizes().base_ptr(ctx, generator),
dims_buf_sz,
int1.const_zero(),
);
generator,
&size_type.const_int(offset as u64, false),
None,
)
};
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(),
);
ctx.builder.build_store(pbuffer, int8.const_int(byte as u64, false)).unwrap();
}
// llvm_intrinsics::call_memcpy_generic(
// ctx,
// buffer.base_ptr(ctx, generator),
// llvm_arg.dim_sizes().base_ptr(ctx, generator),
// 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)
} else {