expm1: Align data type in computation with class definition (#832)

Resolve the FP64 issue in the expm1 operator on ARC.
intel · Aug 29, 2024 · 52c8079 · 52c8079
1 parent dddcf0d
commit 52c8079
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ATen/native/xpu/sycl/UnaryKernels.cpp b/src/ATen/native/xpu/sycl/UnaryKernels.cpp
@@ -234,8 +234,8 @@ struct Expm1Functor {
 template <typename T>
 struct Expm1Functor<c10::complex<T>> {
   c10::complex<T> operator()(c10::complex<T> x) const {
-    auto a = std::sin(.5 * x.imag());
-    auto re = std::expm1(x.real()) * std::cos(x.imag()) - 2 * a * a;
+    auto a = std::sin(T(.5) * x.imag());
+    auto re = std::expm1(x.real()) * std::cos(x.imag()) - T(2) * a * a;
     auto im = std::exp(x.real()) * std::sin(x.imag());
     return c10::complex<T>(re, im);
   }