[XPU] Add bool type for concat op (#10527)

PaddlePaddle · Jun 26, 2024 · bd60e69 · bd60e69
1 parent 79b0a58
commit bd60e69
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/lite/kernels/xpu/concat_compute.cc b/lite/kernels/xpu/concat_compute.cc
@@ -104,6 +104,8 @@ using concati64 =
     paddle::lite::kernels::xpu::ConcatCompute<int64_t, PRECISION(kFloat)>;
 using concati8 =
     paddle::lite::kernels::xpu::ConcatCompute<int8_t, PRECISION(kInt8)>;
+using concatbool =
+    paddle::lite::kernels::xpu::ConcatCompute<bool, PRECISION(kFloat)>;
 
 REGISTER_LITE_KERNEL(concat, kXPU, kFloat, kNCHW, concatfp32, def)
     .BindInput("X", {LiteType::GetTensorTy(TARGET(kXPU), PRECISION(kFloat))})
@@ -147,3 +149,9 @@ REGISTER_LITE_KERNEL(concat, kXPU, kInt8, kNCHW, concati8, concat_INT8)
                {LiteType::GetTensorTy(TARGET(kXPU), PRECISION(kInt32))})
     .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kXPU), PRECISION(kInt8))})
     .Finalize();
+REGISTER_LITE_KERNEL(concat, kXPU, kFloat, kNCHW, concatbool, concat_BOOL)
+    .BindInput("X", {LiteType::GetTensorTy(TARGET(kXPU), PRECISION(kBool))})
+    .BindInput("AxisTensor",
+               {LiteType::GetTensorTy(TARGET(kXPU), PRECISION(kInt32))})
+    .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kXPU), PRECISION(kBool))})
+    .Finalize();