From 5afab2cd1f66684a5721505eea55b4902db43420 Mon Sep 17 00:00:00 2001 From: hellolllw <127483090+hellolllw@users.noreply.github.com> Date: Sun, 12 Mar 2023 20:03:45 +0800 Subject: [PATCH] add register of is_empty (#51484) * Update interpreter_util.cc * Update is_empty_kernel.cc * Update is_empty_kernel.cc --- .../framework/new_executor/interpreter/interpreter_util.cc | 1 - paddle/phi/kernels/is_empty_kernel.cc | 4 +++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc b/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc index f162478344c..b61e3170645 100644 --- 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 OpsNeedSetOutputDtypeWhenRegisterPhiKernel = { "group_norm", "histogram", "instance_norm", - "is_empty", "kthvalue", "lamb", "layer_norm", diff --git a/paddle/phi/kernels/is_empty_kernel.cc b/paddle/phi/kernels/is_empty_kernel.cc index 26c2f978005..4b86f2dfe69 100644 --- 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, -- GitLab