fix the branch of code choose (#31200)

4 years ago · f114c3f8ca
parent d11602481c
commit f114c3f8ca
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/elementwise/elementwise_add_op.h
+++ b/paddle/fluid/operators/elementwise/elementwise_add_op.h
@ -317,7 +317,7 @@ class ElementwiseAddGradKernel : public ElemwiseGradKernel<T> {
 // TODO(@wangchaochaohu, zhouwei35): Fix conv_transpose2d API(dataformat NHWC)
 // error in Windows
-#if defined(PADDLE_WITH_CUDA) && defined(_LINUX)
+#if defined(PADDLE_WITH_CUDA) && !defined(_WIN32)
 #ifdef __NVCC__
    int axis = ctx.Attr<int>("axis");