未验证 提交 ef0eb526 编写于 作者: 张春乔 提交者: GitHub

[phi] add register of angle (#51314)

* delete something in interpreter_util.cc

* rollback

* add REGISTER of angle

* bug fix

* fix something

* Update paddle/phi/kernels/gpu/angle_kernel.cu

* Update paddle/phi/kernels/cpu/angle_kernel.cc

* Update paddle/phi/kernels/gpu/angle_kernel.cu

* Update paddle/phi/kernels/cpu/angle_kernel.cc
上级 19001c6c
...@@ -55,7 +55,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = { ...@@ -55,7 +55,6 @@ static std::set<std::string> OpsNeedSetOutputDtypeWhenRegisterPhiKernel = {
"adamw", "adamw",
"all_close", "all_close",
"all_raw", "all_raw",
"angle",
"any_raw", "any_raw",
"arg_sort", "arg_sort",
"atan2", "atan2",
......
...@@ -26,4 +26,6 @@ PD_REGISTER_KERNEL(angle, ...@@ -26,4 +26,6 @@ PD_REGISTER_KERNEL(angle,
float, float,
double, double,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {
kernel->OutputAt(0).SetDataType(phi::DataType::UNDEFINED);
}
...@@ -26,4 +26,6 @@ PD_REGISTER_KERNEL(angle, ...@@ -26,4 +26,6 @@ PD_REGISTER_KERNEL(angle,
float, float,
double, double,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<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.
先完成此消息的编辑!
想要评论请 注册