zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 12:49:44 +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
1 changed files with 1 additions and 1 deletions
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,