未验证 提交 256103a5 编写于 作者: W wangchaochaohu 提交者: GitHub

[cherry pick]Fix windows error (#31207)

cherry-pick #31068
上级 158fc99b
...@@ -315,7 +315,9 @@ class ElementwiseAddGradKernel : public ElemwiseGradKernel<T> { ...@@ -315,7 +315,9 @@ class ElementwiseAddGradKernel : public ElemwiseGradKernel<T> {
// skip out // skip out
auto *out = dout; 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(_WIN32)
#ifdef __NVCC__ #ifdef __NVCC__
int axis = ctx.Attr<int>("axis"); int axis = ctx.Attr<int>("axis");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册