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/src/Core/arch/AVX512 | |
| parent | 8731452b9777585362b72ca4ae6f7159e07b7ee0 (diff) | |
BF16 for scalar_cmp_with_cast_op
Diffstat (limited to 'Eigen/src/Core/arch/AVX512')
| -rw-r--r-- | Eigen/src/Core/arch/AVX512/PacketMath.h | 3 |
1 files changed, 3 insertions, 0 deletions
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 }; }; |