Add atan(y,x)

See also #443
JuliaGPU · Jan 16, 2020 · def82ed · def82ed
1 parent 1da2e28
commit def82ed
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/device/cuda/math.jl b/src/device/cuda/math.jl
@@ -35,6 +35,8 @@
 # ! CUDAnative.atan2 is equivalent to Base.atan
 @inline atan2(x::Float64, y::Float64) = ccall("extern __nv_atan2", llvmcall, Cdouble, (Cdouble, Cdouble), x, y)
 @inline atan2(x::Float32, y::Float32) = ccall("extern __nv_atan2f", llvmcall, Cfloat, (Cfloat, Cfloat), x, y)
+@inline atan(x::Float64, y::Float64) = atan2(x, y)
+@inline atan(x::Float32, y::Float32) = atan2(x, y)
 
 @inline angle(x::ComplexF64) = atan2(x.im, x.re)
 @inline angle(x::ComplexF32) = atan2(x.im, x.re)