add output defs for eigvals kernel (#51393)

7c9ccf5f · Wang Xin · GitHub · 0f79444e · 7c9ccf5f · 7c9ccf5f
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/cpu/eigvals_kernel.cc paddle/phi/kernels/cpu/eigvals_kernel.cc +3 -1

未找到文件。
--- a/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc
+++ b/paddle/fluid/framework/new_executor/interpreter/interpreter_util.cc
@@ -69,7 +69,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
    "eig",
    "eig_grad",
    "eigh",
-    "eigvals",
    "ftt_c2r",
    "ftt_r2c",
    "fused_adam",

--- a/paddle/phi/kernels/cpu/eigvals_kernel.cc
+++ b/paddle/phi/kernels/cpu/eigvals_kernel.cc
@@ -257,4 +257,6 @@ PD_REGISTER_KERNEL(eigvals,
                   float,
                   double,
                   phi::dtype::complex<float>,
-                   phi::dtype::complex<double>) {}
+                   phi::dtype::complex<double>) {
+  kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::UNDEFINED);
+}