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

Fix typo in row_reduce_small (#2179)

Browse Source
This commit is contained in:
Cheng 2025-05-14 12:19:54 +09:00 committed by GitHub
parent eca2f3eb97
commit 0751263dec
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
mlx/backend/metal/kernels/reduction/reduce_row.h
View File

@ -224,7 +224,7 @@ template <
if ((non_row_reductions < 32 && row_size <= 8) || non_row_reductions <= 8) { if ((non_row_reductions < 32 && row_size <= 8) || non_row_reductions <= 8) {
// Simple loop over non_row_reductions and reduce the row in the thread. // Simple loop over non_row_reductions and reduce the row in the thread.
IdxT out_idx = tid.x + tsize.y * IdxT(tid.y); IdxT out_idx = tid.x + tsize.x * IdxT(tid.y);
in += elem_to_loc<IdxT>(out_idx, shape, strides, ndim); in += elem_to_loc<IdxT>(out_idx, shape, strides, ndim);
for (uint r = 0; r < non_row_reductions; r++) { for (uint r = 0; r < non_row_reductions; r++) {