From 968868fa260da9b50bff97104f8892af9d74b682 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Tue, 29 Jul 2025 17:17:47 -0700 Subject: [PATCH] format --- mlx/backend/cuda/logsumexp.cu | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mlx/backend/cuda/logsumexp.cu b/mlx/backend/cuda/logsumexp.cu index a85b51583..2afcc7e70 100644 --- a/mlx/backend/cuda/logsumexp.cu +++ b/mlx/backend/cuda/logsumexp.cu @@ -54,7 +54,8 @@ __global__ void logsumexp(const T* in, T* out, int axis_size) { // https://github.com/NVIDIA/online-softmax normalizer = normalizer * softmax_exp(prevmax - maxval); for (int i = 0; i < N_READS; i++) { - normalizer = normalizer + softmax_exp(static_cast(vals[i]) - maxval); + normalizer = + normalizer + softmax_exp(static_cast(vals[i]) - maxval); } }