From 497eb948fa3ea3adfb7c1e890a144b58d46eeae8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E6=98=8E=E5=86=AC?= <78149749+winter-wang@users.noreply.github.com> Date: Thu, 22 Sep 2022 16:13:32 +0800 Subject: [PATCH] [NPU] fix CI error in new executor. test=develop (#46292) --- paddle/fluid/operators/share_buffer_op.cc | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/paddle/fluid/operators/share_buffer_op.cc b/paddle/fluid/operators/share_buffer_op.cc index 89d6b231d7..14f6499049 100644 --- a/paddle/fluid/operators/share_buffer_op.cc +++ b/paddle/fluid/operators/share_buffer_op.cc @@ -66,3 +66,11 @@ REGISTER_OPERATOR(share_buffer, ops::ShareBufferOp, ops::ShareBufferOpMaker); // dtype is not important REGISTER_OP_CPU_KERNEL(share_buffer, ops::ShareBufferOpKernel); + +#ifdef PADDLE_WITH_ASCEND_CL +REGISTER_OP_NPU_KERNEL(share_buffer, ops::ShareBufferOpKernel); +#endif + +#ifdef PADDLE_WITH_XPU +REGISTER_OP_XPU_KERNEL(share_buffer, ops::ShareBufferOpKernel); +#endif -- GitLab