forked from M-Labs/nac3
1
0
Fork 0

core/ndstrides: add NDArrayObject::to_any

This commit is contained in:
lyken 2024-08-20 23:23:30 +08:00
parent 3426e0c9c2
commit 1de3f05734
No known key found for this signature in database
GPG Key ID: 3BD5FC6AC8325DD8
1 changed files with 16 additions and 1 deletions

View File

@ -28,7 +28,10 @@ use crate::{
model::*, model::*,
CodeGenContext, CodeGenerator, CodeGenContext, CodeGenerator,
}, },
toplevel::{helper::extract_ndims, numpy::unpack_ndarray_var_tys}, toplevel::{
helper::{create_ndims, extract_ndims},
numpy::{make_ndarray_ty, unpack_ndarray_var_tys},
},
typecheck::typedef::{Type, TypeEnum}, typecheck::typedef::{Type, TypeEnum},
}; };
@ -105,6 +108,18 @@ impl<'ctx> NDArrayObject<'ctx> {
Int(SizeT).const_int(generator, ctx, self.ndims, false) Int(SizeT).const_int(generator, ctx, self.ndims, false)
} }
/// Get the typechecker ndarray type of this [`NDArrayObject`].
pub fn get_type(&self, ctx: &mut CodeGenContext<'ctx, '_>) -> Type {
let ndims = create_ndims(&mut ctx.unifier, self.ndims);
make_ndarray_ty(&mut ctx.unifier, &ctx.primitives, Some(self.dtype), Some(ndims))
}
/// Forget that this is an ndarray and convert into an [`AnyObject`].
pub fn to_any(&self, ctx: &mut CodeGenContext<'ctx, '_>) -> AnyObject<'ctx> {
let ty = self.get_type(ctx);
AnyObject { value: self.instance.value.as_basic_value_enum(), ty }
}
/// Allocate an ndarray on the stack given its `ndims` and `dtype`. /// Allocate an ndarray on the stack given its `ndims` and `dtype`.
/// ///
/// `shape` and `strides` will be automatically allocated onto the stack. /// `shape` and `strides` will be automatically allocated onto the stack.