[XPU] c_sync_calc_stream support more types (#53389)

paddle/fluid/operators/collective/c_sync_calc_stream_op_xpu.cc

@@ -17,5 +17,12 @@ limitations under the License. */
 namespace ops = paddle::operators;
 namespace plat = paddle::platform;
 
-PD_REGISTER_STRUCT_KERNEL(
-    c_sync_calc_stream, XPU, ALL_LAYOUT, ops::CSyncCalcStreamKernel, float) {}
+PD_REGISTER_STRUCT_KERNEL(c_sync_calc_stream,
+                          XPU,
+                          ALL_LAYOUT,
+                          ops::CSyncCalcStreamKernel,
+                          float,
+                          double,
+                          int,
+                          int64_t,
+                          plat::float16) {}

paddle/phi/backends/xpu/xpu2_op_list.cc

@@ -115,7 +115,12 @@ XPUOpMap& get_kl2_ops() {
        XPUKernelSet({phi::DataType::FLOAT16,
                      phi::DataType::FLOAT32,
                      phi::DataType::INT32})},
-      {"c_sync_calc_stream", XPUKernelSet({phi::DataType::FLOAT32})},
+      {"c_sync_calc_stream",
+       XPUKernelSet({phi::DataType::FLOAT16,
+                     phi::DataType::FLOAT32,
+                     phi::DataType::FLOAT64,
+                     phi::DataType::INT32,
+                     phi::DataType::INT64})},
       {"c_sync_comm_stream", XPUKernelSet({phi::DataType::FLOAT32})},
       {"cast",
        XPUKernelSet({phi::DataType::FLOAT32,