diff options
author | 2016-04-01 13:32:29 +0100 | |
---|---|---|
committer | 2016-04-01 13:32:29 +0100 | |
commit | dd5d390daf3a3a561a772b64f1b602e5f240bf8b (patch) | |
tree | c68be25e3a95d9c840fad61498628473a79ff6b9 /Eigen/src/Core/arch/CUDA/MathFunctions.h | |
parent | af4ef540bfeb381daaae86f91d492eed39f84e68 (diff) |
Added zeta function.
Diffstat (limited to 'Eigen/src/Core/arch/CUDA/MathFunctions.h')
-rw-r--r-- | Eigen/src/Core/arch/CUDA/MathFunctions.h | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/Eigen/src/Core/arch/CUDA/MathFunctions.h b/Eigen/src/Core/arch/CUDA/MathFunctions.h index 6822700f8..858775523 100644 --- a/Eigen/src/Core/arch/CUDA/MathFunctions.h +++ b/Eigen/src/Core/arch/CUDA/MathFunctions.h @@ -91,6 +91,20 @@ double2 pdigamma<double2>(const double2& a) using numext::digamma; return make_double2(digamma(a.x), digamma(a.y)); } + +template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE +float4 pzeta<float4>(const float4& a) +{ + using numext::zeta; + return make_float4(zeta(a.x), zeta(a.y), zeta(a.z), zeta(a.w)); +} + +template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE +double2 pzeta<double2>(const double2& a) +{ + using numext::zeta; + return make_double2(zeta(a.x), zeta(a.y)); +} template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE float4 perf<float4>(const float4& a) |