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

[core] codegen/expr: Add concrete ndims value to error message

Browse Source
This commit is contained in:
David Mak 2025-02-10 10:37:55 +08:00
parent 69542c38a2
commit 67f42185de
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nac3core/src/codegen/expr.rs
View File

@ -43,7 +43,7 @@ use super::{
use crate::{ use crate::{
symbol_resolver::{SymbolValue, ValueEnum}, symbol_resolver::{SymbolValue, ValueEnum},
toplevel::{ toplevel::{
helper::{arraylike_flatten_element_type, PrimDef}, helper::{arraylike_flatten_element_type, extract_ndims, PrimDef},
numpy::unpack_ndarray_var_tys, numpy::unpack_ndarray_var_tys,
DefinitionId, TopLevelDef, DefinitionId, TopLevelDef,
}, },
@ -1775,10 +1775,13 @@ pub fn gen_unaryop_expr_with_values<'ctx, G: CodeGenerator>(
if op == ast::Unaryop::Invert { if op == ast::Unaryop::Invert {
ast::Unaryop::Not ast::Unaryop::Not
} else { } else {
let ndims = extract_ndims(&ctx.unifier, ty);
codegen_unreachable!( codegen_unreachable!(
ctx, ctx,
"ufunc {} not supported for ndarray[bool, N]", "ufunc {} not supported for ndarray[bool, {}]",
op.op_info().method_name, op.op_info().method_name,
ndims,
) )
} }
} else { } else {