zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-26 02:33:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix build (#1693)

Browse Source
This commit is contained in:
Awni Hannun 2024-12-11 23:56:25 -08:00 committed by GitHub
parent 4e1e9520e1
commit 3a21f61772
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/src/ops.cpp
View File

@ -105,7 +105,7 @@ void init_ops(nb::module_& m) {
)pbdoc"); )pbdoc");
m.def( m.def(
"unflatten", "unflatten",
&unflatten, &mx::unflatten,
nb::arg(), nb::arg(),
"axis"_a, "axis"_a,
"shape"_a, "shape"_a,