[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.