From 364cfa268660d79bdbb7074342b423a9b82c2d2f Mon Sep 17 00:00:00 2001 From: wangchaochaohu Date: Tue, 23 Feb 2021 12:27:45 +0800 Subject: [PATCH] fix windows for optimization of elementwise_add Op (#31068) * fix windows for optimization of elementwise_add Op --- paddle/fluid/operators/elementwise/elementwise_add_op.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/elementwise/elementwise_add_op.h b/paddle/fluid/operators/elementwise/elementwise_add_op.h index 41e97a3946..8c1279a579 100644 --- 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 { // 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("axis"); -- GitLab