add register of bincount (#51508)

* Update interpreter_util.cc * Update bincount_kernel.cc * Update bincount_kernel.cu

add register of bincount (#51508)
* Update interpreter_util.cc * Update bincount_kernel.cc * Update bincount_kernel.cu
48090c72 · hellockx · GitHub · 5afab2cd · 48090c72 · 48090c72
3 changed file
--- a/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc
+++ b/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc
@@ -57,7 +57,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
    "arg_sort",
    "atan2",
    "auc",
-    "bincount",
    "clip_by_norm",
    "complex",
    "conv3d_coo",

--- a/paddle/phi/kernels/cpu/bincount_kernel.cc
+++ b/paddle/phi/kernels/cpu/bincount_kernel.cc
@@ -111,4 +111,6 @@ PD_REGISTER_KERNEL(bincount,
                   float,
                   double,
                   int,
-                   int64_t) {}
+                   int64_t) {
+  kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
+}
--- a/paddle/phi/kernels/gpu/bincount_kernel.cu
+++ b/paddle/phi/kernels/gpu/bincount_kernel.cu
@@ -158,4 +158,6 @@ PD_REGISTER_KERNEL(bincount,
                   float,
                   double,
                   int,
-                   int64_t) {}
+                   int64_t) {
+  kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
+}