M-Labs/nac3
M-Labs/nac3
8
1
Fork 12
Code Issues 99 Pull Requests 19 Releases Wiki Activity

core/ndstrides: fix alloca_ndarray comment

Browse Source
This commit is contained in:
lyken 2024-07-30 15:09:14 +08:00
parent ac7cc15d90
commit 7e056b9747
1 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
nac3core/src/codegen/numpy_new/util.rs
View File

@ -17,18 +17,16 @@ use crate::{
typecheck::typedef::{Type, TypeEnum}, typecheck::typedef::{Type, TypeEnum},
}; };
/** /// Allocate an ndarray on the stack given its `ndims`.
Allocate an ndarray on the stack given its `ndims`. ///
/// `shape` and `strides` will be automatically allocated on the stack.
`shape` and `strides` will be automatically allocated on the stack. ///
/// The returned ndarray's content will be:
The returned ndarray's content will be: /// - `data`: `nullptr`
- `data`: `nullptr` /// - `itemsize`: **uninitialized** value
- `itemsize`: **uninitialized** value /// - `ndims`: initialized value, set to the input `ndims`
- `ndims`: initialized value, set to the input `ndims` /// - `shape`: initialized pointer to an allocated stack with **uninitialized** values
- `shape`: initialized pointer to an allocated stack with **uninitialized** values /// - `strides`: initialized pointer to an allocated stack with **uninitialized** values
- `strides`: initialized pointer to an allocated stack with **uninitialized** values
*/
pub fn alloca_ndarray<'ctx, G>( pub fn alloca_ndarray<'ctx, G>(
generator: &mut G, generator: &mut G,
ctx: &mut CodeGenContext<'ctx, '_>, ctx: &mut CodeGenContext<'ctx, '_>,