WIP trait implementations

Author: sbastrakov

include/alpaka/math/asin/AsinUniformCudaHipBuiltIn.hpp

@@ -59,7 +59,7 @@ namespace alpaka
                 {
                     // This holds everywhere, including the branch cuts: asin(z) = i * ln(sqrt(1 - z^2) - i * z)
                     return Complex<T>(0.0, 1.0)
-                        * log(acos_ctx, sqrt(acos_ctx, T(1.0) - arg * arg) - Complex<T>(0.0, 1.0) * arg);
+                        * log(asin_ctx, sqrt(asin_ctx, T(1.0) - arg * arg) - Complex<T>(0.0, 1.0) * arg);
                 }
             };
         } // namespace traits

include/alpaka/math/atan/AtanUniformCudaHipBuiltIn.hpp

@@ -58,7 +58,7 @@ namespace alpaka
                 {
                     // This holds everywhere, including the branch cuts: atan(z) = -i/2 * ln((i - z) / (i + z))
                     return Complex<T>{0.0, -0.5}
-                    * log(acos_ctx, (Complex<T>{0.0, 1.0} + arg) / (Complex<T>{0.0, 1.0} + arg));
+                    * log(atan_ctx, (Complex<T>{0.0, 1.0} + arg) / (Complex<T>{0.0, 1.0} + arg));
                 }
             };
         } // namespace traits

include/alpaka/math/cos/CosUniformCudaHipBuiltIn.hpp

@@ -57,11 +57,12 @@ namespace alpaka
                 __device__ auto operator()(CosUniformCudaHipBuiltIn const& cos_ctx, Complex<T> const& arg)
                 {
                     // cos(z) = 0.5 * (exp(i * z) + exp(-i * z))
-                    return T(0.5) * (exp(cos_ctx, Complex<T>{0.0, 1.0) * arg) + exp(cos_ctx, Complex<T>{0.0, -1.0) * arg));
+                    return T(0.5)
+                        * (exp(cos_ctx, Complex<T>{0.0, 1.0} * arg) + exp(cos_ctx, Complex<T>{0.0, -1.0} * arg));
                 }
             };
-                } // namespace traits
-            } // namespace math
+        } // namespace traits
+    } // namespace math
 } // namespace alpaka
 
 #endif

include/alpaka/math/exp/ExpUniformCudaHipBuiltIn.hpp

@@ -52,7 +52,7 @@ namespace alpaka
                 }
             };
             //! The CUDA/HIP exp Complex<T> specialization.
-            template<template T>
+            template<typename T>
             struct Exp<ExpUniformCudaHipBuiltIn, Complex<T>>
             {
                 __device__ auto operator()(ExpUniformCudaHipBuiltIn const& exp_ctx, Complex<T> const& arg)

include/alpaka/math/rsqrt/RsqrtUniformCudaHipBuiltIn.hpp

@@ -55,7 +55,7 @@ namespace alpaka
             {
                 __device__ auto operator()(RsqrtUniformCudaHipBuiltIn const& rsqrt_ctx, Complex<T> const& arg)
                 {
-                    return T{1.0} / sqrt(rsqrt_ctx, arg) l
+                    return T{1.0} / sqrt(rsqrt_ctx, arg);
                 }
             };
         } // namespace traits

include/alpaka/math/sin/SinUniformCudaHipBuiltIn.hpp

@@ -57,11 +57,12 @@ namespace alpaka
                 __device__ auto operator()(SinUniformCudaHipBuiltIn const& sin_ctx, Complex<T> const& arg)
                 {
                     // sin(z) = (exp(i * z) - exp(-i * z)) / 2i
-                    return (exp(cos_ctx, Complex<T>{0.0, 1.0) * arg) - exp(cos_ctx, Complex<T>{0.0, -1.0) * arg)) / Complex<T>{0.0, 2.0};
+                    return (exp(sin_ctx, Complex<T>{0.0, 1.0} * arg) - exp(sin_ctx, Complex<T>{0.0, -1.0} * arg))
+                        / Complex<T>{0.0, 2.0};
                 }
             };
-                } // namespace traits
-            } // namespace math
+        } // namespace traits
+    } // namespace math
 } // namespace alpaka
 
 #endif

include/alpaka/math/tan/TanUniformCudaHipBuiltIn.hpp

@@ -59,7 +59,7 @@ namespace alpaka
                     // tan(z) = i * (e^-iz - e^iz) / (e^-iz + e^iz)
                     auto const exp1 = exp(tan_ctx, Complex<T>{0.0, -1.0} * arg);
                     auto const exp2 = exp(tan_ctx, Complex<T>{0.0, 1.0} * arg);
-                    return Complex{0.0, 1.0} * (exp1 - exp2) / (exp1 + exp2);
+                    return Complex<T>{0.0, 1.0} * (exp1 - exp2) / (exp1 + exp2);
                 }
             };
         } // namespace traits