diff options
| author |  Sheng Yang <yang.sheng@intel.com> | 2020-07-01 18:33:42 +0000 |
|---|---|---|
| committer |  Rasmus Munk Larsen <rmlarsen@google.com> | 2020-07-01 18:33:42 +0000 |
| commit | 116c5235ac5600672982e5a12a0ab89d19d2d5f0 (patch) | |
| tree | 2b16c8ec16042549cb3f4acc2e0f20809dd92752 /Eigen | |
| parent | 8731452b9777585362b72ca4ae6f7159e07b7ee0 (diff) | |
BF16 for scalar_cmp_with_cast_op
Diffstat (limited to 'Eigen')
| -rw-r--r-- | Eigen/src/Core/GenericPacketMath.h | 3 | ||||
| -rw-r--r-- | Eigen/src/Core/arch/AVX/PacketMath.h | 2 | ||||
| -rw-r--r-- | Eigen/src/Core/arch/AVX512/PacketMath.h | 3 | ||||
| -rwxr-xr-x | Eigen/src/Core/arch/SSE/PacketMath.h | 2 | ||||
| -rw-r--r-- | Eigen/src/Core/functors/BinaryFunctors.h | 2 |
5 files changed, 11 insertions, 1 deletions
diff --git a/Eigen/src/Core/GenericPacketMath.h b/Eigen/src/Core/GenericPacketMath.h index 39bba577a..ebae1a34f 100644 --- a/Eigen/src/Core/GenericPacketMath.h +++ b/Eigen/src/Core/GenericPacketMath.h @@ -58,6 +58,9 @@ struct default_packet_traits HasConj = 1, HasSetLinear = 1, HasBlend = 0, + // This flag is used to indicate whether packet comparison is supported. + // pcmp_eq, pcmp_lt and pcmp_le should be defined for it to be true. + HasCmp = 0, HasDiv = 0, HasSqrt = 0, diff --git a/Eigen/src/Core/arch/AVX/PacketMath.h b/Eigen/src/Core/arch/AVX/PacketMath.h index b27a62cbc..b17c34015 100644 --- a/Eigen/src/Core/arch/AVX/PacketMath.h +++ b/Eigen/src/Core/arch/AVX/PacketMath.h @@ -63,6 +63,7 @@ template<> struct packet_traits<float> : default_packet_traits size = 8, HasHalfPacket = 1, + HasCmp = 1, HasDiv = 1, HasSin = EIGEN_FAST_MATH, HasCos = EIGEN_FAST_MATH, @@ -93,6 +94,7 @@ template<> struct packet_traits<double> : default_packet_traits size=4, HasHalfPacket = 1, + HasCmp = 1, HasDiv = 1, HasExp = 1, HasSqrt = 1, diff --git a/Eigen/src/Core/arch/AVX512/PacketMath.h b/Eigen/src/Core/arch/AVX512/PacketMath.h index ed15e0889..f55a50596 100644 --- a/Eigen/src/Core/arch/AVX512/PacketMath.h +++ b/Eigen/src/Core/arch/AVX512/PacketMath.h @@ -103,6 +103,7 @@ template<> struct packet_traits<float> : default_packet_traits HasTanh = EIGEN_FAST_MATH, HasErf = EIGEN_FAST_MATH, #endif + HasCmp = 1, HasDiv = 1 }; }; @@ -119,6 +120,7 @@ template<> struct packet_traits<double> : default_packet_traits HasSqrt = EIGEN_FAST_MATH, HasRsqrt = EIGEN_FAST_MATH, #endif + HasCmp = 1, HasDiv = 1 }; }; @@ -1656,6 +1658,7 @@ struct packet_traits<bfloat16> : default_packet_traits { HasTanh = EIGEN_FAST_MATH, HasErf = EIGEN_FAST_MATH, #endif + HasCmp = 1, HasDiv = 1 }; }; diff --git a/Eigen/src/Core/arch/SSE/PacketMath.h b/Eigen/src/Core/arch/SSE/PacketMath.h index 9f2e922a2..1d640cf76 100755 --- a/Eigen/src/Core/arch/SSE/PacketMath.h +++ b/Eigen/src/Core/arch/SSE/PacketMath.h @@ -94,6 +94,7 @@ struct packet_traits<float> : default_packet_traits { size = 4, HasHalfPacket = 0, + HasCmp = 1, HasDiv = 1, HasSin = EIGEN_FAST_MATH, HasCos = EIGEN_FAST_MATH, @@ -128,6 +129,7 @@ struct packet_traits<double> : default_packet_traits { size=2, HasHalfPacket = 0, + HasCmp = 1, HasDiv = 1, HasExp = 1, HasSqrt = 1, diff --git a/Eigen/src/Core/functors/BinaryFunctors.h b/Eigen/src/Core/functors/BinaryFunctors.h index 54bcddd34..d8b7b1eba 100644 --- a/Eigen/src/Core/functors/BinaryFunctors.h +++ b/Eigen/src/Core/functors/BinaryFunctors.h @@ -260,7 +260,7 @@ template<typename LhsScalar, typename RhsScalar, ComparisonName cmp> struct functor_traits<scalar_cmp_with_cast_op<LhsScalar,RhsScalar, cmp> > { enum { Cost = (NumTraits<LhsScalar>::AddCost+NumTraits<RhsScalar>::AddCost)/2, - PacketAccess = internal::is_same<LhsScalar, RhsScalar>::value && internal::is_same<LhsScalar, float>::value + PacketAccess = internal::is_same<LhsScalar, RhsScalar>::value && packet_traits<LhsScalar>::HasCmp }; }; |