未验证 提交 7c9ccf5f 编写于 作者: W Wang Xin 提交者: GitHub

add output defs for eigvals kernel (#51393)

上级 0f79444e
......@@ -69,7 +69,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
"eig",
"eig_grad",
"eigh",
"eigvals",
"ftt_c2r",
"ftt_r2c",
"fused_adam",
......
......@@ -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);
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册