Add digamma for CPU + CUDA. Includes tests.

2026-04-10 11:34:33 +08:00 · 2015-12-24 21:15:38 -08:00
parent bdcbc66a5c
commit f7362772e3
9 changed files with 447 additions and 62 deletions
--- a/Eigen/src/Core/arch/CUDA/PacketMath.h
+++ b/Eigen/src/Core/arch/CUDA/PacketMath.h
@@ -40,6 +40,7 @@ template<> struct packet_traits<float> : default_packet_traits
    HasSqrt = 1,
    HasRsqrt = 1,
    HasLGamma = 1,
+    HasDiGamma = 1,
    HasErf = 1,
    HasErfc = 1,

@@ -63,6 +64,7 @@ template<> struct packet_traits<double> : default_packet_traits
    HasSqrt = 1,
    HasRsqrt = 1,
    HasLGamma = 1,
+    HasDiGamma = 1,
    HasErf = 1,
    HasErfc = 1,