fix windows for optimization of elementwise_add Op (#31068)

* fix windows for optimization of elementwise_add Op
4 years ago · 364cfa2686
parent 781df300d0
commit 364cfa2686
1 changed files with 3 additions and 1 deletions
--- a/paddle/fluid/operators/elementwise/elementwise_add_op.h
+++ b/paddle/fluid/operators/elementwise/elementwise_add_op.h
@ -315,7 +315,9 @@ class ElementwiseAddGradKernel : public ElemwiseGradKernel<T> {
    // skip out
    auto *out = dout;

-#ifdef PADDLE_WITH_CUDA
+// TODO(@wangchaochaohu, zhouwei35): Fix conv_transpose2d API(dataformat NHWC)
+// error in Windows
+#if defined(PADDLE_WITH_CUDA) && defined(_LINUX)
 #ifdef __NVCC__

    int axis = ctx.Attr<int>("axis");