diff options
| author |  Antonio Sanchez <cantonios@google.com> | 2021-04-27 14:12:11 -0700 |
|---|---|---|
| committer | Antonio Sanchez <cantonios@google.com> | 2021-04-27 14:12:11 -0700 |
| commit | 172db7bfc32def5ed0f885287e352b63dd5cd767 (patch) | |
| tree | b362124663b240406c84124cea215974413d2c2f /Eigen | |
| parent | 2ced0cc233fff6ef16c4d098b03aeeb69ff7c509 (diff) | |
Add missing pcmp_lt_or_nan for NEON Packet4bf.
Diffstat (limited to 'Eigen')
| -rw-r--r-- | Eigen/src/Core/arch/NEON/PacketMath.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/Eigen/src/Core/arch/NEON/PacketMath.h b/Eigen/src/Core/arch/NEON/PacketMath.h index 9cf4e0712..2b48570d1 100644 --- a/Eigen/src/Core/arch/NEON/PacketMath.h +++ b/Eigen/src/Core/arch/NEON/PacketMath.h @@ -3701,6 +3701,11 @@ template<> EIGEN_STRONG_INLINE Packet4bf pcmp_lt<Packet4bf>(const Packet4bf& a, return F32MaskToBf16Mask(pcmp_lt<Packet4f>(Bf16ToF32(a), Bf16ToF32(b))); } +template<> EIGEN_STRONG_INLINE Packet4bf pcmp_lt_or_nan<Packet4bf>(const Packet4bf& a, const Packet4bf& b) +{ + return F32MaskToBf16Mask(pcmp_lt_or_nan<Packet4f>(Bf16ToF32(a), Bf16ToF32(b))); +} + template<> EIGEN_STRONG_INLINE Packet4bf pcmp_le<Packet4bf>(const Packet4bf& a, const Packet4bf& b) { return F32MaskToBf16Mask(pcmp_le<Packet4f>(Bf16ToF32(a), Bf16ToF32(b))); |