Merge pull request #516 from PhilipVinc/PhilipVinc-patch-1

fix CuArrays.jl #543 - complex exp type instability
JuliaGPU · Dec 23, 2019 · 9497d08 · 9497d08
2 parents 2e83fc6 + 25854c3
commit 9497d08
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/device/cuda/math.jl b/src/device/cuda/math.jl
@@ -113,8 +113,8 @@
 @inline ldexp(x::Float32, y::Int32) = @wrap __nv_ldexpf(x::float, y::i32)::float
 
 @inline exp(x::Complex{Float64}) = exp(x.re) * (cos(x.im) + 1.0im * sin(x.im))
-@inline exp(x::Complex{Float32}) = exp(x.re) * (cos(x.im) + 1.0im * sin(x.im))
-@inline exp_fast(x::Complex{Float32}) = exp_fast(x.re) * (cos_fast(x.im) + 1.0im * sin_fast(x.im))
+@inline exp(x::Complex{Float32}) = exp(x.re) * (cos(x.im) + 1.0f0im * sin(x.im))
+@inline exp_fast(x::Complex{Float32}) = exp_fast(x.re) * (cos_fast(x.im) + 1.0f0im * sin_fast(x.im))
 
 ## error