未验证 提交 d431b7c7 编写于 作者: D Difer 提交者: GitHub

resgister for ftt_r2c, ftt_c2_r (#51563)

* resgister for ftt_r2c, ftt_c2_r

* fix clang-format
上级 545e20f8
......@@ -57,8 +57,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
"clip_by_norm",
"eig_grad",
"eigh",
"ftt_c2r",
"ftt_r2c",
"graph_sample_neighbors",
"group_norm",
"lamb",
......
......@@ -27,6 +27,9 @@ PD_REGISTER_KERNEL(fft_c2r,
ALL_LAYOUT,
phi::FFTC2RKernel,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {}
phi::dtype::complex<double>) {
kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
}
PD_REGISTER_KERNEL(fft_r2c, CPU, ALL_LAYOUT, phi::FFTR2CKernel, float, double) {
kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
}
......@@ -27,6 +27,9 @@ PD_REGISTER_KERNEL(fft_c2r,
ALL_LAYOUT,
phi::FFTC2RKernel,
phi::dtype::complex<float>,
phi::dtype::complex<double>) {}
phi::dtype::complex<double>) {
kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
}
PD_REGISTER_KERNEL(fft_r2c, GPU, ALL_LAYOUT, phi::FFTR2CKernel, float, double) {
kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册