fix windows for optimization of elementwise_add Op (#31068)

* fix windows for optimization of elementwise_add Op

fix windows for optimization of elementwise_add Op (#31068)
* fix windows for optimization of elementwise_add Op
364cfa26 · wangchaochaohu · GitHub · 781df300 · 364cfa26
显示空白变更内容
内联并排

Showing with 3 addition and 1 deletion

paddle/fluid/operators/elementwise/elementwise_add_op.h paddle/fluid/operators/elementwise/elementwise_add_op.h +3 -1

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