add output defs for nonzero and nms (#51325)

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

@@ -94,8 +94,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
     "multiclass_nms3",
     "multinomial",
     "nanmedian",
-    "nms",
-    "nonzero",
     "numl",
     "qr",
     "qr_grad",

paddle/phi/kernels/cpu/nms_kernel.cc

@@ -94,4 +94,6 @@ void NMSKernel(const Context& dev_ctx,
 
 }  // namespace phi
 
-PD_REGISTER_KERNEL(nms, CPU, ALL_LAYOUT, phi::NMSKernel, float, double) {}
+PD_REGISTER_KERNEL(nms, CPU, ALL_LAYOUT, phi::NMSKernel, float, double) {
+  kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::INT64);
+}

paddle/phi/kernels/cpu/nonzero_kernel.cc

@@ -92,4 +92,6 @@ PD_REGISTER_KERNEL(nonzero,
                    int16_t,
                    bool,
                    float,
-                   double) {}
+                   double) {
+  kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::INT64);
+}

paddle/phi/kernels/gpu/nms_kernel.cu

@@ -113,4 +113,6 @@ void NMSKernel(const Context& dev_ctx,
                      dev_ctx.stream());
 }
 }  // namespace phi
-PD_REGISTER_KERNEL(nms, GPU, ALL_LAYOUT, phi::NMSKernel, float, double) {}
+PD_REGISTER_KERNEL(nms, GPU, ALL_LAYOUT, phi::NMSKernel, float, double) {
+  kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::INT64);
+}

paddle/phi/kernels/gpu/nonzero_kernel.cu

@@ -83,4 +83,6 @@ PD_REGISTER_KERNEL(nonzero,
                    int16_t,
                    bool,
                    float,
-                   double) {}
+                   double) {
+  kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::INT64);
+}

paddle/phi/kernels/xpu/nonzero_kernel.cc

@@ -69,4 +69,6 @@ void NonZeroKernel(const Context& dev_ctx,
 }  // namespace phi
 
 PD_REGISTER_KERNEL(
-    nonzero, XPU, ALL_LAYOUT, phi::NonZeroKernel, int, bool, float) {}
+    nonzero, XPU, ALL_LAYOUT, phi::NonZeroKernel, int, bool, float) {
+  kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::INT64);
+}