[XPU] add int type for concat and split functor (#50200)

paddle/fluid/pybind/process_group_utils.h

@@ -143,6 +143,12 @@ void ConcatDenseTensorWithType(const phi::XPUContext &dev_ctx,
     case phi::DataType::FLOAT32:
       ConcatDenseTensor<phi::XPUContext, float>()(dev_ctx, t_list, p_out);
       break;
+    case phi::DataType::INT32:
+      ConcatDenseTensor<phi::XPUContext, int32_t>()(dev_ctx, t_list, p_out);
+      break;
+    case phi::DataType::INT64:
+      ConcatDenseTensor<phi::XPUContext, int64_t>()(dev_ctx, t_list, p_out);
+      break;
     default:
       PADDLE_THROW(platform::errors::Unimplemented(
           "Data type (%s) is not supported when it concats tensors.", type));
@@ -205,6 +211,12 @@ void SplitDenseTensorWithType(const phi::XPUContext &dev_ctx,
     case phi::DataType::FLOAT32:
       SplitDenseTensor<phi::XPUContext, float>()(dev_ctx, t_in, p_list);
       break;
+    case phi::DataType::INT32:
+      SplitDenseTensor<phi::XPUContext, int32_t>()(dev_ctx, t_in, p_list);
+      break;
+    case phi::DataType::INT64:
+      SplitDenseTensor<phi::XPUContext, int64_t>()(dev_ctx, t_in, p_list);
+      break;
     default:
       PADDLE_THROW(platform::errors::Unimplemented(
           "Data type (%s) is not supported when it splits tensors.", type));

paddle/phi/kernels/xpu/concat_and_split_functor.cc

@@ -127,6 +127,8 @@ class SplitFunctor<XPUContext, T> {
 
 DEFINE_XPU_FUNCTOR(float)
 DEFINE_XPU_FUNCTOR(phi::dtype::float16)
+DEFINE_XPU_FUNCTOR(int32_t)
+DEFINE_XPU_FUNCTOR(int64_t)
 
 }  // namespace funcs
 }  // namespace phi