From 8ba1b0f41a7950dc3e1d4ed75859e36c73311235 Mon Sep 17 00:00:00 2001 From: David Tellenbach Date: Thu, 13 Aug 2020 15:48:40 +0000 Subject: bfloat16 packetmath for Arm Neon backend --- Eigen/src/Core/arch/NEON/MathFunctions.h | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'Eigen/src/Core/arch/NEON/MathFunctions.h') 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(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 -- cgit v1.2.3