Merge branch 'release/2.5' into xyt/gamma_double_r25

intel · Aug 29, 2024 · 2b7d7ff · 2b7d7ff
2 parents e9ad734 + 2bd8ce9
commit 2b7d7ff
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);
   }