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

spelling: across

Browse Source 
Signed-off-by: Josh Soref <2119212+jsoref@users.noreply.github.com>
This commit is contained in:
Josh Soref
2024-01-01 22:27:22 -05:00
parent 647f80d09c
commit 73272bb87d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/backend/metal/kernels/arg_reduce.metal
View File

@@ -114,7 +114,7 @@ template <typename T, typename Op, int N_READS>
// 4. Reduce among them and go to 3 // 4. Reduce among them and go to 3
// 4. Reduce in each simd_group // 4. Reduce in each simd_group
// 6. Write in the thread local memory // 6. Write in the thread local memory
// 6. Reduce them accross thread group // 6. Reduce them across thread group
// 7. Write the output without need for atomic // 7. Write the output without need for atomic
Op op; Op op;