1
0
forked from M-Labs/nac3

core: Fix IRRT argument order to ndarray_flatten_index

This commit is contained in:
David Mak 2024-02-15 15:08:10 +08:00
parent 8c7e44098a
commit 1eacaf9afa

View File

@ -732,10 +732,10 @@ pub fn call_ndarray_flatten_index<'ctx>(
let ndarray_flatten_index_fn = ctx.module.get_function(ndarray_flatten_index_fn_name).unwrap_or_else(|| { let ndarray_flatten_index_fn = ctx.module.get_function(ndarray_flatten_index_fn_name).unwrap_or_else(|| {
let fn_type = llvm_usize.fn_type( let fn_type = llvm_usize.fn_type(
&[ &[
llvm_pusize.into(),
llvm_usize.into(), llvm_usize.into(),
llvm_pusize.into(),
llvm_pi32.into(), llvm_pi32.into(),
llvm_pusize.into(), llvm_usize.into(),
], ],
false, false,
); );
@ -752,10 +752,10 @@ pub fn call_ndarray_flatten_index<'ctx>(
.build_call( .build_call(
ndarray_flatten_index_fn, ndarray_flatten_index_fn,
&[ &[
ndarray_num_dims.into(),
ndarray_dims.get_ptr(ctx).into(), ndarray_dims.get_ptr(ctx).into(),
ndarray_num_dims.into(),
indices.into(),
indices_size.into(), indices_size.into(),
indices_data.get_ptr(ctx).into(),
], ],
"", "",
) )