Skip to content

Commit 256103a

Browse files
[cherry pick]Fix windows error (#31207)
cherry-pick #31068
1 parent 158fc99 commit 256103a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

paddle/fluid/operators/elementwise/elementwise_add_op.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -315,7 +315,9 @@ class ElementwiseAddGradKernel : public ElemwiseGradKernel<T> {
315315
// skip out
316316
auto *out = dout;
317317

318-
#ifdef PADDLE_WITH_CUDA
318+
// TODO(@wangchaochaohu, zhouwei35): Fix conv_transpose2d API(dataformat NHWC)
319+
// error in Windows
320+
#if defined(PADDLE_WITH_CUDA) && !defined(_WIN32)
319321
#ifdef __NVCC__
320322

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

0 commit comments

Comments
 (0)