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 5aee8681d1
1 changed files with 34 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
nac3artiq/src/codegen.rs
View File

@ -30,7 +30,7 @@ use pyo3::{
use crate::{symbol_resolver::InnerResolver, timeline::TimeFns};
use inkwell::types::BasicType;
use itertools::Itertools;
use itertools::{repeat_n, Itertools};
use nac3core::codegen::classes::{ArrayLikeIndexer, ArrayLikeValue, ArraySliceValue, NDArrayType};
use nac3core::codegen::llvm_intrinsics;
use std::{
@ -431,7 +431,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))),
@ -581,23 +582,38 @@ fn rpc_codegen_callback_fn<'ctx>(
let buffer = ctx.builder.build_array_alloca(int8, buffer_size, "").unwrap();
let buffer = ArraySliceValue::from_ptr_val(buffer, buffer_size, None);
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(),
);
for (offset, byte) in [0u8, 0u8, 0u8, 0u8, 129u8, 129u8, 0u8, 0u8, 0u8, 0u8, 0u8, 0u8].into_iter().enumerate()
// repeat_n(129u8, 12).into_iter().enumerate()
{
let pbuffer = unsafe {
buffer.ptr_offset_unchecked(
ctx,
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 {