mirror of
https://gitlab.com/libeigen/eigen.git
synced 2026-04-10 11:34:33 +08:00
Added zeta function.
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -40,6 +40,7 @@ template<> struct packet_traits<float> : default_packet_traits
|
||||
HasRsqrt = 1,
|
||||
HasLGamma = 1,
|
||||
HasDiGamma = 1,
|
||||
HasZeta = 1,
|
||||
HasErf = 1,
|
||||
HasErfc = 1,
|
||||
HasIgamma = 1,
|
||||
|
||||
Reference in New Issue
Block a user