diff --git a/paddle/phi/kernels/gpudnn/softmax_gpudnn.h b/paddle/phi/kernels/gpudnn/softmax_gpudnn.h index 8ba329301c9160cd6daf70f524484315a6cb0eb8..9db814f19864343f35a32d38c19f7da3f9b97e38 100644 --- a/paddle/phi/kernels/gpudnn/softmax_gpudnn.h +++ b/paddle/phi/kernels/gpudnn/softmax_gpudnn.h @@ -499,8 +499,7 @@ __global__ void KeMatrixSoftmaxForward(T* softmax, const T* src, int dim_size) { // write data to softmax_output according to the LogMode if (LogMode) { - LogSoftmaxForwardFunctor reduction(thread_max, - std::log(thread_exp)); + LogSoftmaxForwardFunctor reduction(thread_max, thread_exp); if (input_align_shift == output_align_shift) { ThreadVecWriteVec( batch_output, batch_input, dim_size, input_align_shift, reduction);