fix windows for optimization of elementwise_add Op (#31068)

* fix windows for optimization of elementwise_add Op
PaddlePaddle · Feb 23, 2021 · 364cfa2 · 364cfa2
1 parent 781df30
commit 364cfa2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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");