diff --git a/mlx/backend/metal/kernels/reduce.metal b/mlx/backend/metal/kernels/reduce.metal index 85ff41f44..4182184c2 100644 --- a/mlx/backend/metal/kernels/reduce.metal +++ b/mlx/backend/metal/kernels/reduce.metal @@ -65,7 +65,7 @@ template in += grid_size * N_READS; } - // Sepate case for the last set as we close the reduction size + // Separate case for the last set as we close the reduction size size_t curr_idx = (gid + r * (size_t)grid_size) * N_READS; if (curr_idx < in_size) { int max_reads = in_size - curr_idx;