diff --git a/paddle/fluid/operators/elementwise/elementwise_mod_op_xpu.cc b/paddle/fluid/operators/elementwise/elementwise_mod_op_xpu.cc index 43fadce98f3f7a..156589384c0dd2 100644 --- a/paddle/fluid/operators/elementwise/elementwise_mod_op_xpu.cc +++ b/paddle/fluid/operators/elementwise/elementwise_mod_op_xpu.cc @@ -43,7 +43,7 @@ namespace ops = paddle::operators; REGISTER_OP_XPU_KERNEL(elementwise_mod, ops::ElementwiseModXPUKernel, ops::ElementwiseModXPUKernel, - ops::ElementwiseModXPUKernel, - ops::ElementwiseModXPUKernel); + ops::ElementwiseModXPUKernel, + ops::ElementwiseModXPUKernel); #endif diff --git a/paddle/fluid/platform/device/xpu/xpu1_op_list.h b/paddle/fluid/platform/device/xpu/xpu1_op_list.h index a76bdd4ae96798..94cf71e9a6000c 100644 --- a/paddle/fluid/platform/device/xpu/xpu1_op_list.h +++ b/paddle/fluid/platform/device/xpu/xpu1_op_list.h @@ -102,6 +102,8 @@ XPUOpMap& get_kl1_ops() { XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, {"elementwise_min_grad", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, + {"elementwise_mod", + XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, {"elementwise_min", XPUKernelSet({pOpKernelType(vartype::FP32, XPUPlace())})}, {"elementwise_mul_grad",