Add output defs for sgd kernel (#51332)

* Add output defs for sgd kernel

* add datatype infer for sgd

* add infer logic

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

@@ -96,7 +96,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
     "select",
     "send_recv",
     "send_ue_recv",
-    "sgd",
     "svd",
     "sync_batch_norm_grad",
     "unique",

paddle/phi/infermeta/multiary.cc

@@ -2484,6 +2484,15 @@ void SgdInferMeta(const MetaTensor& param,
 
   param_out->set_dims(param.dims());
   param_out->set_dtype(param.dtype());
+  if (multi_precision) {
+    master_param_out->set_dims(master_param.dims());
+    if (DataType::FLOAT16 == master_param.dtype() ||
+        DataType::BFLOAT16 == master_param.dtype()) {
+      master_param_out->set_dtype(DataType::FLOAT32);
+    } else {
+      master_param_out->set_dtype(master_param.dtype());
+    }
+  }
 }
 
 void SendUERecvInferMeta(const MetaTensor& x,

paddle/phi/kernels/gpu/sgd_kernel.cu

@@ -187,7 +187,9 @@ PD_REGISTER_KERNEL(sgd,
                    phi::SGDDenseKernel,
                    phi::dtype::float16,
                    float,
-                   double) {}
+                   double) {
+  kernel->OutputAt(1).SetDataType(phi::DataType::UNDEFINED);
+}
 
 PD_REGISTER_KERNEL(sgd_dense_param_sparse_grad,
                    GPU,