fix datatype error

add op to in xpu1_op_list *test=kunlun
PaddlePaddle · Apr 25, 2022 · f2c4337 · f2c4337
1 parent dd90c72
commit f2c4337
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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<float>,
                        ops::ElementwiseModXPUKernel<paddle::platform::float16>,
-                       ops::ElementwiseModXPUKernel<int>,
-                       ops::ElementwiseModXPUKernel<int64>);
+                       ops::ElementwiseModXPUKernel<int32_t>,
+                       ops::ElementwiseModXPUKernel<int64_t>);
 
 #endif
diff --git 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",