【NPU】Add int dtype kernel for reshape2 op (#31864)

* fix * fix
PaddlePaddle · Mar 25, 2021 · 9754d0a · 9754d0a · paddle-bot-old · Mar 25, 2021
1 parent 48aa922
commit 9754d0a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/paddle/fluid/operators/reshape_op_npu.cc b/paddle/fluid/operators/reshape_op_npu.cc
@@ -67,10 +67,20 @@ namespace ops = paddle::operators;
 
 REGISTER_OP_NPU_KERNEL(
     reshape2, ops::Reshape2NPUKernel<paddle::platform::NPUDeviceContext, float>,
+    ops::Reshape2NPUKernel<paddle::platform::NPUDeviceContext, int>,
+    ops::Reshape2NPUKernel<paddle::platform::NPUDeviceContext, int64_t>,
+    ops::Reshape2NPUKernel<paddle::platform::NPUDeviceContext, bool>,
+    ops::Reshape2NPUKernel<paddle::platform::NPUDeviceContext, double>,
+    ops::Reshape2NPUKernel<paddle::platform::NPUDeviceContext, uint8_t>,
     ops::Reshape2NPUKernel<paddle::platform::NPUDeviceContext,
                            paddle::platform::float16>);
 REGISTER_OP_NPU_KERNEL(
     reshape2_grad,
     ops::Reshape2GradNPUKernel<paddle::platform::NPUDeviceContext, float>,
+    ops::Reshape2GradNPUKernel<paddle::platform::NPUDeviceContext, int>,
+    ops::Reshape2GradNPUKernel<paddle::platform::NPUDeviceContext, int64_t>,
+    ops::Reshape2GradNPUKernel<paddle::platform::NPUDeviceContext, bool>,
+    ops::Reshape2GradNPUKernel<paddle::platform::NPUDeviceContext, double>,
+    ops::Reshape2GradNPUKernel<paddle::platform::NPUDeviceContext, uint8_t>,
     ops::Reshape2GradNPUKernel<paddle::platform::NPUDeviceContext,
                                paddle::platform::float16>);