M-Labs/nac3
M-Labs
/
nac3
8
0
Fork 10
Code Issues 100 Pull Requests 20 Releases Wiki Activity

core: improve comments in type_inferencer/mod.rs

This commit is contained in:
lyken 2024-06-27 14:46:48 +08:00
parent 5b11a1dbdd
commit 9808923258
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nac3core/src/typecheck/type_inferencer/mod.rs
View File

@ -880,7 +880,7 @@ impl<'a> Inferencer<'a> {
// Check `shape_ty` to see if its a list of int32s, a tuple of int32s, or just int32. // Check `shape_ty` to see if its a list of int32s, a tuple of int32s, or just int32.
// Otherwise throw an error as that would mean the user wrote an ill-typed `shape_expr`. // Otherwise throw an error as that would mean the user wrote an ill-typed `shape_expr`.
// //
// Here, we also take the opportunity to deduce `ndims` statically for 2. and 3. // Here, we also take the opportunity to deduce `ndims` statically.
let shape_ty_enum = &*self.unifier.get_ty(shape_ty); let shape_ty_enum = &*self.unifier.get_ty(shape_ty);
let ndims = match shape_ty_enum { let ndims = match shape_ty_enum {
TypeEnum::TList { ty } => { TypeEnum::TList { ty } => {
@ -1292,7 +1292,7 @@ impl<'a> Inferencer<'a> {
{ {
let shape_expr = args.remove(0); let shape_expr = args.remove(0);
let (ndims, shape) = let (ndims, shape) =
self.fold_numpy_function_call_shape_argument(*id, 0, shape_expr)?; // Special handling the `shape` self.fold_numpy_function_call_shape_argument(*id, 0, shape_expr)?; // Special handling for `shape`
let ndims = self.unifier.get_fresh_literal(vec![SymbolValue::U64(ndims)], None); let ndims = self.unifier.get_fresh_literal(vec![SymbolValue::U64(ndims)], None);
let ret = make_ndarray_ty( let ret = make_ndarray_ty(