未验证 提交 364cfa26 编写于 作者: W wangchaochaohu 提交者: GitHub

fix windows for optimization of elementwise_add Op (#31068)

* fix windows for optimization of elementwise_add Op
上级 781df300
...@@ -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(_LINUX)
#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.
先完成此消息的编辑!
想要评论请 注册