zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 17:31:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Pass missing stream arg in array.flatten (#1111)

Browse Source
This commit is contained in:
Cheng 2024-05-14 21:50:16 +08:00 committed by GitHub
parent bf925d9dc7
commit 56a4eaed72
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/array.cpp
View File

@ -1139,7 +1139,7 @@ void init_array(nb::module_& m) {
int start_axis, int start_axis,
int end_axis, int end_axis,
const StreamOrDevice& s) { const StreamOrDevice& s) {
return flatten(a, start_axis, end_axis); return flatten(a, start_axis, end_axis, s);
}, },
"start_axis"_a = 0, "start_axis"_a = 0,
"end_axis"_a = -1, "end_axis"_a = -1,