diff options
Diffstat (limited to 'Eigen')
-rw-r--r-- | Eigen/Core | 1 | ||||
-rw-r--r-- | Eigen/src/Core/arch/AVX512/PacketMath.h | 13 |
2 files changed, 8 insertions, 6 deletions
diff --git a/Eigen/Core b/Eigen/Core index e6ef4abbe..fa908ac01 100644 --- a/Eigen/Core +++ b/Eigen/Core @@ -316,6 +316,7 @@ using std::ptrdiff_t; #include "src/Core/arch/SSE/PacketMath.h" #include "src/Core/arch/AVX/PacketMath.h" #include "src/Core/arch/AVX512/PacketMath.h" + #include "src/Core/arch/AVX512/MathFunctions.h" #elif defined EIGEN_VECTORIZE_AVX // Use AVX for floats and doubles, SSE for integers #include "src/Core/arch/SSE/PacketMath.h" diff --git a/Eigen/src/Core/arch/AVX512/PacketMath.h b/Eigen/src/Core/arch/AVX512/PacketMath.h index 55d93e35b..302f46736 100644 --- a/Eigen/src/Core/arch/AVX512/PacketMath.h +++ b/Eigen/src/Core/arch/AVX512/PacketMath.h @@ -54,11 +54,12 @@ template<> struct packet_traits<float> : default_packet_traits AlignedOnScalar = 1, size = 16, HasHalfPacket = 1, - HasExp = 0, + HasLog = 1, + HasExp = 1, HasDiv = 1, HasBlend = 1, - HasSqrt = 0, - HasRsqrt = 0, + HasSqrt = 1, + HasRsqrt = 1, HasSelect = 1, HasEq = 1 }; @@ -72,11 +73,11 @@ template<> struct packet_traits<double> : default_packet_traits AlignedOnScalar = 1, size = 8, HasHalfPacket = 1, - HasExp = 0, + HasExp = 1, HasDiv = 1, HasBlend = 1, - HasSqrt = 0, - HasRsqrt = 0, + HasSqrt = 1, + HasRsqrt = EIGEN_FAST_MATH, HasSelect = 1, HasEq = 1 }; |