add output defs for send_u_recv (#51501)

paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc

@@ -91,7 +91,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
     "rnn",
     "search_sort",
     "select",
-    "send_recv",
     "send_ue_recv",
     "sync_batch_norm_grad",
     "unique",

paddle/phi/kernels/cpu/send_u_recv_kernel.cc

@@ -184,4 +184,6 @@ PD_REGISTER_KERNEL(send_u_recv,
                    float,
                    double,
                    int,
-                   int64_t) {}
+                   int64_t) {
+  kernel->OutputAt(1).SetDataType(phi::DataType::INT32);
+}

paddle/phi/kernels/gpu/send_u_recv_kernel.cu

@@ -196,4 +196,6 @@ PD_REGISTER_KERNEL(send_u_recv,
                    double,
                    int,
                    int64_t,
-                   phi::dtype::float16) {}
+                   phi::dtype::float16) {
+  kernel->OutputAt(1).SetDataType(phi::DataType::INT32);
+}