add register of kthvalue (#51534)

* add register of KthvalueKernel

add register of KthvalueKernel

* Update kthvalue_kernel.cc

* Update kthvalue_kernel.cu

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

@@ -71,7 +71,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
     "group_norm",
     "histogram",
     "instance_norm",
-    "kthvalue",
     "lamb",
     "layer_norm",
     "layer_norm_grad",

paddle/phi/kernels/cpu/kthvalue_kernel.cc

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

paddle/phi/kernels/gpu/kthvalue_kernel.cu

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