From 23504a61848368ec1bf8ecb1aa18d6ddb4017d50 Mon Sep 17 00:00:00 2001 From: zhouwei25 Date: Fri, 17 Jun 2022 11:59:14 +0000 Subject: [PATCH] fix comment --- paddle/phi/kernels/sparse/gpu/softmax_kernel.cu | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/phi/kernels/sparse/gpu/softmax_kernel.cu b/paddle/phi/kernels/sparse/gpu/softmax_kernel.cu index a3bd396dd9c72..6353a46eea1ef 100644 --- a/paddle/phi/kernels/sparse/gpu/softmax_kernel.cu +++ b/paddle/phi/kernels/sparse/gpu/softmax_kernel.cu @@ -56,9 +56,9 @@ __global__ void SoftmaxGpuKernel(const IntT* x_crows, if (idx >= row_nnz) break; auto functor = phi::funcs::CudaExpFunctor(); - out_values[row_first + idx] = - functor(x_values[row_first + idx] - row_max_val); - exp_sum += functor(x_values[row_first + idx] - row_max_val); + T exp = functor(x_values[row_first + idx] - row_max_val); + exp_sum += exp; + out_values[row_first + idx] = exp; } T row_exp_sum = phi::funcs::warpReduceSum(exp_sum, 0xFFFFFFFF);