diff --git a/mlx/backend/metal/kernels/arg_reduce.metal b/mlx/backend/metal/kernels/arg_reduce.metal index 31bcbfa05..467e768d6 100644 --- a/mlx/backend/metal/kernels/arg_reduce.metal +++ b/mlx/backend/metal/kernels/arg_reduce.metal @@ -114,7 +114,7 @@ template // 4. Reduce among them and go to 3 // 4. Reduce in each simd_group // 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 Op op;