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

Resolved ambiguity in mlx::core::take_along_axis (#1822)

Browse Source 
* Resolved ambiguity in mlx::core::take_along_axis

Detected by GCC 10 on riscv64-linux-gnu.

* Formatted

* Removed superfluous parentheses in random_tests.cpp
This commit is contained in:
Jesper Stemann Andersen
2025-02-04 15:06:17 +01:00
committed by GitHub
parent 1156c84e86
commit f6c0499b8d
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/ops.cpp
View File

@@ -3081,7 +3081,8 @@ array take_along_axis(
axis = axis < 0 ? a.ndim() + axis : axis;
// Broadcast indices and input ignoring the take axis
auto inputs = broadcast_arrays({a, indices}, {axis - int(a.ndim())}, s);
auto inputs =
broadcast_arrays({a, indices}, std::vector<int>{axis - int(a.ndim())}, s);
auto out_shape = inputs[1].shape();
return array(