zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-18 10:26:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix shapeless export (#2148)

Browse Source
This commit is contained in:
Awni Hannun
2025-05-01 15:02:02 -07:00
committed by GitHub
parent a3a632d567
commit 9daa6b003f
2 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/export.cpp
View File

@@ -470,6 +470,9 @@ bool FunctionTable::match(
if (x.dtype() != y.dtype()) {
return false;
}
if (x.ndim() != y.ndim()) {
return false;
}
if (!shapeless && x.shape() != y.shape()) {
return false;
}