zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 01:41:17 +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 end_axis,
const StreamOrDevice& s) {
return flatten(a, start_axis, end_axis);
return flatten(a, start_axis, end_axis, s);
},
"start_axis"_a = 0,
"end_axis"_a = -1,