diff options
author | David Tellenbach <david.tellenbach@me.com> | 2020-08-13 15:48:40 +0000 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2020-08-13 15:48:40 +0000 |
commit | 8ba1b0f41a7950dc3e1d4ed75859e36c73311235 (patch) | |
tree | 662ac7a45607d0527c925f22c78f0739aa9823d9 /Eigen/src/Core/arch/NEON/MathFunctions.h | |
parent | 704798d1df4866be335ca013da19a44791f85a7e (diff) |
bfloat16 packetmath for Arm Neon backend
Diffstat (limited to 'Eigen/src/Core/arch/NEON/MathFunctions.h')
-rw-r--r-- | Eigen/src/Core/arch/NEON/MathFunctions.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/Eigen/src/Core/arch/NEON/MathFunctions.h b/Eigen/src/Core/arch/NEON/MathFunctions.h index 1c025618e..ab549a683 100644 --- a/Eigen/src/Core/arch/NEON/MathFunctions.h +++ b/Eigen/src/Core/arch/NEON/MathFunctions.h @@ -38,6 +38,12 @@ template<> EIGEN_DEFINE_FUNCTION_ALLOWING_MULTIPLE_DEFINITIONS EIGEN_UNUSED Pack template<> EIGEN_DEFINE_FUNCTION_ALLOWING_MULTIPLE_DEFINITIONS EIGEN_UNUSED Packet4f ptanh<Packet4f>(const Packet4f& x) { return internal::generic_fast_tanh_float(x); } +BF16_PACKET_FUNCTION(Packet4f, Packet4bf, psin) +BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pcos) +BF16_PACKET_FUNCTION(Packet4f, Packet4bf, plog) +BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pexp) +BF16_PACKET_FUNCTION(Packet4f, Packet4bf, ptanh) + } // end namespace internal } // end namespace Eigen |