diff --git a/mlx/backend/cuda/reduce/col_reduce.cu b/mlx/backend/cuda/reduce/col_reduce.cu index 63940b2d4..6b18d1d0a 100644 --- a/mlx/backend/cuda/reduce/col_reduce.cu +++ b/mlx/backend/cuda/reduce/col_reduce.cu @@ -339,13 +339,12 @@ void col_reduce( // Small col reduce with a single or contiguous reduction axis if (args.non_col_reductions == 1 && args.reduction_size <= 32 && args.reduction_stride % (16 / in.itemsize()) == 0) { - col_reduce_small( - encoder, in, out, reduce_type, axes, plan, std::move(args)); + col_reduce_small(encoder, in, out, reduce_type, axes, plan, args); return; } // Fallback col reduce - col_reduce_looped(encoder, in, out, reduce_type, axes, plan, std::move(args)); + col_reduce_looped(encoder, in, out, reduce_type, axes, plan, args); } } // namespace mlx::core