10 #ifndef EIGEN_MATH_FUNCTIONS_CUDA_H
11 #define EIGEN_MATH_FUNCTIONS_CUDA_H
20 #if defined(__CUDACC__) && defined(EIGEN_USE_GPU)
21 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
22 float4 plog<float4>(
const float4& a)
24 return make_float4(logf(a.x), logf(a.y), logf(a.z), logf(a.w));
27 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
28 double2 plog<double2>(
const double2& a)
30 return make_double2(log(a.x), log(a.y));
33 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
34 float4 pexp<float4>(
const float4& a)
36 return make_float4(expf(a.x), expf(a.y), expf(a.z), expf(a.w));
39 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
40 double2 pexp<double2>(
const double2& a)
42 return make_double2(exp(a.x), exp(a.y));
45 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
46 float4 psqrt<float4>(
const float4& a)
48 return make_float4(sqrtf(a.x), sqrtf(a.y), sqrtf(a.z), sqrtf(a.w));
51 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
52 double2 psqrt<double2>(
const double2& a)
54 return make_double2(sqrt(a.x), sqrt(a.y));
57 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
58 float4 prsqrt<float4>(
const float4& a)
60 return make_float4(rsqrtf(a.x), rsqrtf(a.y), rsqrtf(a.z), rsqrtf(a.w));
63 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
64 double2 prsqrt<double2>(
const double2& a)
66 return make_double2(rsqrt(a.x), rsqrt(a.y));
75 #endif // EIGEN_MATH_FUNCTIONS_CUDA_H
Definition: Eigen_Colamd.h:54