From 0be820cb0cab1e4f0b08c539343ed3dfb5911d38 Mon Sep 17 00:00:00 2001 From: Tao Luo Date: Wed, 5 Feb 2020 19:44:05 +0800 Subject: [PATCH] fix sigmoid cudnn bug (#22439) (#22449) Co-authored-by: Manjunath Bhat --- paddle/fluid/operators/activation_cudnn_op.cu.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/paddle/fluid/operators/activation_cudnn_op.cu.cc b/paddle/fluid/operators/activation_cudnn_op.cu.cc index f03355eb44..84767f7c22 100644 --- a/paddle/fluid/operators/activation_cudnn_op.cu.cc +++ b/paddle/fluid/operators/activation_cudnn_op.cu.cc @@ -171,10 +171,10 @@ class CudnnActivationGradKernel namespace plat = paddle::platform; namespace ops = paddle::operators; -#define FOR_EACH_CUDNN_OP_FUNCTOR(__macro) \ - __macro(relu, CudnnReluFunctor, CudnnReluGradFunctor); \ - __macro(relu6, CudnnRelu6Functor, CudnnRelu6GradFunctor); \ - __macro(sigmoid, CudnnTanhFunctor, CudnnTanhGradFunctor); \ +#define FOR_EACH_CUDNN_OP_FUNCTOR(__macro) \ + __macro(relu, CudnnReluFunctor, CudnnReluGradFunctor); \ + __macro(relu6, CudnnRelu6Functor, CudnnRelu6GradFunctor); \ + __macro(sigmoid, CudnnSigmoidFunctor, CudnnSigmoidGradFunctor); \ __macro(tanh, CudnnTanhFunctor, CudnnTanhGradFunctor) #define REGISTER_ACTIVATION_CUDNN_KERNEL(act_type, functor, grad_functor) \ -- GitLab