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));
69 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
70 float4 plgamma<float4>(
const float4& a)
72 return make_float4(lgammaf(a.x), lgammaf(a.y), lgammaf(a.z), lgammaf(a.w));
75 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
76 double2 plgamma<double2>(
const double2& a)
78 return make_double2(lgamma(a.x), lgamma(a.y));
81 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
82 float4 perf<float4>(
const float4& a)
84 return make_float4(erf(a.x), erf(a.y), erf(a.z), erf(a.w));
87 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
88 double2 perf<double2>(
const double2& a)
90 return make_double2(erf(a.x), erf(a.y));
93 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
94 float4 perfc<float4>(
const float4& a)
96 return make_float4(erfc(a.x), erfc(a.y), erfc(a.z), erfc(a.w));
99 template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
100 double2 perfc<double2>(
const double2& a)
102 return make_double2(erfc(a.x), erfc(a.y));
112 #endif // EIGEN_MATH_FUNCTIONS_CUDA_H
Definition: Eigen_Colamd.h:54