[XPU] add registration of SplitWithNumKernel with int64. (#54478)

paddle/phi/backends/xpu/xpu2_op_list.cc

@@ -726,7 +726,8 @@ XPUOpMap& get_kl2_ops() {
       {"split_with_num",
        XPUKernelSet({phi::DataType::FLOAT32,
                      phi::DataType::FLOAT16,
-                     phi::DataType::INT32})},
+                     phi::DataType::INT32,
+                     phi::DataType::INT64})},
       {"sqrt", XPUKernelSet({phi::DataType::FLOAT32, phi::DataType::FLOAT16})},
       {"sqrt_grad", XPUKernelSet({phi::DataType::FLOAT32})},
       {"square_grad",

paddle/phi/kernels/xpu/split_kernel.cc

@@ -79,5 +79,6 @@ PD_REGISTER_KERNEL(split_with_num,
                    ALL_LAYOUT,
                    phi::SplitWithNumKernel,
                    float,
+                   int64_t,
                    int,
                    phi::dtype::float16) {}