Siddhangana/nac3
1
0
Fork 0
forked from M-Labs/nac3
Code Pull Requests Activity

[core] codegen/values/ndarray: Fix Vec allocation

Browse Source
This commit is contained in:
David Mak 2025-01-06 16:31:17 +08:00
parent d9c180ed13
commit 2271b46b96
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nac3core/src/codegen/values/ndarray/mod.rs
View File

@ -950,10 +950,9 @@ impl<'ctx, Index: UntypedArrayLikeAccessor<'ctx>> UntypedArrayLikeMutator<'ctx,
/// This function is used generating strides for globally defined contiguous ndarrays. /// This function is used generating strides for globally defined contiguous ndarrays.
#[must_use] #[must_use]
pub fn make_contiguous_strides(itemsize: u64, ndims: u64, shape: &[u64]) -> Vec<u64> { pub fn make_contiguous_strides(itemsize: u64, ndims: u64, shape: &[u64]) -> Vec<u64> {
let mut strides = Vec::with_capacity(ndims as usize); let mut strides = vec![0u64; ndims as usize];
let mut stride_product = 1u64; let mut stride_product = 1u64;
for i in 0..ndims { for axis in (0..ndims).rev() {
let axis = ndims - i - 1;
strides[axis as usize] = stride_product * itemsize; strides[axis as usize] = stride_product * itemsize;
stride_product *= shape[axis as usize]; stride_product *= shape[axis as usize];
} }