fix sigmoid cudnn bug (#22439)

* Sigmoid bug fix, test=develop * fix code format test=develop Co-authored-by: Manjunath Bhat <manjunathbhat9920@gmail.com>
5 years ago · 943cb8c664
parent 6e4f39a061
commit 943cb8c664
1 changed files with 4 additions and 4 deletions
--- 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) \