未验证 提交 497eb948 编写于 作者: 王明冬 提交者: GitHub

[NPU] fix CI error in new executor. test=develop (#46292)

上级 567e2fc8
...@@ -66,3 +66,11 @@ REGISTER_OPERATOR(share_buffer, ops::ShareBufferOp, ops::ShareBufferOpMaker); ...@@ -66,3 +66,11 @@ REGISTER_OPERATOR(share_buffer, ops::ShareBufferOp, ops::ShareBufferOpMaker);
// dtype is not important // dtype is not important
REGISTER_OP_CPU_KERNEL(share_buffer, ops::ShareBufferOpKernel<float>); REGISTER_OP_CPU_KERNEL(share_buffer, ops::ShareBufferOpKernel<float>);
#ifdef PADDLE_WITH_ASCEND_CL
REGISTER_OP_NPU_KERNEL(share_buffer, ops::ShareBufferOpKernel<float>);
#endif
#ifdef PADDLE_WITH_XPU
REGISTER_OP_XPU_KERNEL(share_buffer, ops::ShareBufferOpKernel<float>);
#endif
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册