fix sigmoid cudnn bug (#22439) (#22449)

Co-authored-by: N Manjunath Bhat <manjunathbhat9920@gmail.com>

fix sigmoid cudnn bug (#22439) (#22449)
Co-authored-by: N Manjunath Bhat <manjunathbhat9920@gmail.com>
0be820cb · Tao Luo · GitHub · fb98116c · 0be820cb
隐藏空白更改
内联并排

Showing with 4 addition and 4 deletion

paddle/fluid/operators/activation_cudnn_op.cu.cc paddle/fluid/operators/activation_cudnn_op.cu.cc +4 -4

未找到文件。
--- 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) \