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

core: check int32 obj_id directly in fold_numpy_function_call_shape_argument

This commit is contained in:
lyken 2024-07-05 10:36:47 +08:00
parent ba32fab374
commit 24ac3820b2
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -980,11 +980,10 @@ impl<'a> Inferencer<'a> {
// `ndims` can be deduced statically from the inferred Tuple type.
tuple_element_types.len() as u64
}
TypeEnum::TObj { .. } => {
// Handle 3. An integer (generalized as [`TypeEnum::TObj`])
// Typecheck
self.unify(self.primitives.int32, shape_ty, &shape.location)?;
TypeEnum::TObj { obj_id, .. }
if *obj_id == self.primitives.int32.obj_id(self.unifier).unwrap() =>
{
// Handle 3. An int32 (generalized as [`TypeEnum::TObj`])
// Deduce `ndims`
1