diff --git a/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc b/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc index fddeebc215317485e484f8824a92e6f94b665531..161ad1a3f3ea5055fc34fba8bd8ab07d7ef5334b 100644 --- a/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc +++ b/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc @@ -56,7 +56,6 @@ static std::set OpsNeedSetOutputDtypeWhenRegisterPhiKernel = { "any_raw", "arg_sort", "atan2", - "auc", "clip_by_norm", "complex", "conv3d_coo", diff --git a/paddle/phi/kernels/cpu/auc_kernel.cc b/paddle/phi/kernels/cpu/auc_kernel.cc index 0cf85348e6a78b3bd58a3ff3e573e69034da0310..647fc592d0e38de042960229e3eac17f6013c79b 100644 --- a/paddle/phi/kernels/cpu/auc_kernel.cc +++ b/paddle/phi/kernels/cpu/auc_kernel.cc @@ -207,4 +207,8 @@ void AucKernel(const Context &dev_ctx, } // namespace phi -PD_REGISTER_KERNEL(auc, CPU, ALL_LAYOUT, phi::AucKernel, float) {} +PD_REGISTER_KERNEL(auc, CPU, ALL_LAYOUT, phi::AucKernel, float) { + kernel->OutputAt(0).SetDataType(phi::DataType::FLOAT64); + kernel->OutputAt(1).SetDataType(phi::DataType::INT64); + kernel->OutputAt(2).SetDataType(phi::DataType::INT64); +} diff --git a/paddle/phi/kernels/gpu/auc_kernel.cu b/paddle/phi/kernels/gpu/auc_kernel.cu index c815f33a6675f27b885d4c068f0c8e2fd315cddb..f733df24cf898e5f272c407991aa31388be702ac 100644 --- a/paddle/phi/kernels/gpu/auc_kernel.cu +++ b/paddle/phi/kernels/gpu/auc_kernel.cu @@ -273,4 +273,8 @@ void AucKernel(const Context &dev_ctx, } // namespace phi -PD_REGISTER_KERNEL(auc, GPU, ALL_LAYOUT, phi::AucKernel, float) {} +PD_REGISTER_KERNEL(auc, GPU, ALL_LAYOUT, phi::AucKernel, float) { + kernel->OutputAt(0).SetDataType(phi::DataType::FLOAT64); + kernel->OutputAt(1).SetDataType(phi::DataType::INT64); + kernel->OutputAt(2).SetDataType(phi::DataType::INT64); +}