add register of is_empty (#51484)

* Update interpreter_util.cc * Update is_empty_kernel.cc * Update is_empty_kernel.cc

add register of is_empty (#51484)
* Update interpreter_util.cc * Update is_empty_kernel.cc * Update is_empty_kernel.cc
5afab2cd · hellolllw · GitHub · 4a4fcc2d · 5afab2cd · 5afab2cd
Showing with 3 addition and 2 deletion

paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc ...id/framework/new_executor/interpreter/interpreter_util.cc +0 -1

paddle/phi/kernels/is_empty_kernel.cc paddle/phi/kernels/is_empty_kernel.cc +3 -1

未找到文件。
--- a/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc
+++ b/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc
@@ -73,7 +73,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
    "group_norm",
    "histogram",
    "instance_norm",
-    "is_empty",
    "kthvalue",
    "lamb",
    "layer_norm",

--- a/paddle/phi/kernels/is_empty_kernel.cc
+++ b/paddle/phi/kernels/is_empty_kernel.cc
@@ -39,7 +39,9 @@ PD_REGISTER_KERNEL(is_empty,
                   float,
                   double,
                   int,
-                   int64_t) {}
+                   int64_t) {
+  kernel->OutputAt(0).SetDataType(phi::DataType::BOOL);
+}

 #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
 PD_REGISTER_KERNEL(is_empty,