diff options
Diffstat (limited to 'Eigen/src/Core/arch/AVX512/PacketMath.h')
| -rw-r--r-- | Eigen/src/Core/arch/AVX512/PacketMath.h | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/Eigen/src/Core/arch/AVX512/PacketMath.h b/Eigen/src/Core/arch/AVX512/PacketMath.h index 76f3366d7..8b946b3e1 100644 --- a/Eigen/src/Core/arch/AVX512/PacketMath.h +++ b/Eigen/src/Core/arch/AVX512/PacketMath.h @@ -1626,8 +1626,6 @@ template <> struct is_arithmetic<Packet16bf> { enum { value = true }; }; template <> struct packet_traits<bfloat16> : default_packet_traits { typedef Packet16bf type; - // There is no half-size packet for current Packet16bf. - // TODO: support as SSE path. typedef Packet8bf half; enum { Vectorizable = 1, @@ -1884,6 +1882,11 @@ EIGEN_STRONG_INLINE Packet16bf pmax<Packet16bf>(const Packet16bf& a, } template <> +EIGEN_STRONG_INLINE Packet16bf plset<Packet16bf>(const bfloat16& a) { + return F32ToBf16(plset<Packet16f>(static_cast<float>(a))); +} + +template <> EIGEN_STRONG_INLINE Packet8bf predux_half_dowto4<Packet16bf>(const Packet16bf& a) { Packet8bf lane0 = _mm256_extractf128_si256(a, 0); Packet8bf lane1 = _mm256_extractf128_si256(a, 1); |