add output defs for atan2 kernel (#51312)

* fix atan2

* fix

* fix

* fix

* fix error

* fix error

* fix

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

@@ -55,7 +55,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
     "adamw",
     "any_raw",
     "arg_sort",
-    "atan2",
     "clip_by_norm",
     "eig_grad",
     "eigh",

paddle/phi/infermeta/binary.cc

@@ -166,6 +166,8 @@ void Atan2InferMeta(const MetaTensor& x, const MetaTensor& y, MetaTensor* out) {
   if (x.dtype() == DataType::INT32 || x.dtype() == DataType::INT64 ||
       y.dtype() == DataType::INT32 || y.dtype() == DataType::INT64) {
     out->set_dtype(DataType::FLOAT64);
+  } else {
+    out->set_dtype(x.dtype());
   }
 }
 

paddle/phi/kernels/cpu/atan2_kernel.cc

@@ -24,4 +24,6 @@ PD_REGISTER_KERNEL(atan2,
                    double,
                    phi::dtype::float16,
                    int,
-                   int64_t) {}
+                   int64_t) {
+  kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
+}

paddle/phi/kernels/gpu/atan2_kernel.cu

@@ -24,4 +24,6 @@ PD_REGISTER_KERNEL(atan2,
                    double,
                    phi::dtype::float16,
                    int,
-                   int64_t) {}
+                   int64_t) {
+  kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
+}