[Complex] Fix wrong reciprocal result when meets complex dtype

[ooooo-create]

PR Category

Operator Mechanism

PR Types

Bug fixes

Description

#73714

Fix wrong reciprocal result when meets complex dtype

[paddle-bot]

你的PR提交成功，感谢你对开源项目的贡献!
请关注后续CI自动化测试结果，详情请参考Paddle-CI手册。
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

[ooooo-create]

/re-run all-failed

[zhwesky2010]

LGTM

[Malone-AI]

What about extracting shared CPU/GPU code?
Like this one

template <typename T> __host__ __device__ __forceinline__ ComplexType<T> reciprocal_complex_kernel( const ComplexType<T>& val) { auto both_inf = [](T real, T imag) { return (isinf(real) && isinf(imag)); }; auto either_inf = [](T real, T imag) { return isinf(real) || isinf(imag); }; auto either_nan = [](T real, T imag) { return isnan(real) || isnan(imag); }; if (either_nan(val.real, val.imag) || both_inf(val.real, val.imag)) { T quiet_nan = std::numeric_limits<T>::quiet_NaN(); return ComplexType<T>(quiet_nan, quiet_nan); } else if (either_inf(val.real, val.imag)) { return ComplexType<T>{static_cast<T>(0), static_cast<T>(0)}; } return static_cast<ComplexType<T>>(1.0) / val; } template <typename T> struct Reciprocal { HOSTDEVICE ComplexType<T> operator()(const ComplexType<T>& val) const { return reciprocal_complex_kernel(val); } }; template <typename T> struct CudaReciprocalFunctor<ComplexType<T>> : public BaseActivationFunctor<ComplexType<T>> { __device__ __forceinline__ ComplexType<T> operator()( const ComplexType<T> x) const { return reciprocal_complex_kernel(x); } };

[luotao1]

What about extracting shared CPU/GPU code?

@Malone-AI, you can create a new PR to extract shared CPU/GPU code.