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

core/ndstrides: add TODO in np_reshape

This commit is contained in:
lyken 2024-07-30 15:40:41 +08:00
parent 3dc4b17310
commit fb54d5d112
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nac3core/src/toplevel/builtins.rs
View File

@ -1341,6 +1341,8 @@ impl<'a> BuiltinBuilder<'a> {
match prim { match prim {
PrimDef::FunNpReshape => { PrimDef::FunNpReshape => {
// TODO: Support scalar inputs, e.g., `np.reshape(99, (1, 1, 1, 1))`
let new_ndim_ty = self.unifier.get_fresh_var(Some("NewNDim".into()), None); let new_ndim_ty = self.unifier.get_fresh_var(Some("NewNDim".into()), None);
let returned_ndarray_ty = make_ndarray_ty( let returned_ndarray_ty = make_ndarray_ty(
self.unifier, self.unifier,