From 85428a344025316db654c65e2628d4ceea1eec47 Mon Sep 17 00:00:00 2001 From: Guoqiang QI <425418567@qq.com> Date: Tue, 8 Sep 2020 09:04:03 +0000 Subject: Add Neon psqrt and pexp --- Eigen/src/Core/arch/NEON/MathFunctions.h | 9 +++++++++ 1 file changed, 9 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 ab549a683..8bea0ac3c 100644 --- a/Eigen/src/Core/arch/NEON/MathFunctions.h +++ b/Eigen/src/Core/arch/NEON/MathFunctions.h @@ -44,6 +44,15 @@ BF16_PACKET_FUNCTION(Packet4f, Packet4bf, plog) BF16_PACKET_FUNCTION(Packet4f, Packet4bf, pexp) BF16_PACKET_FUNCTION(Packet4f, Packet4bf, ptanh) + +//---------- double ---------- + +#if EIGEN_ARCH_ARM64 && !EIGEN_APPLE_DOUBLE_NEON_BUG +template<> EIGEN_DEFINE_FUNCTION_ALLOWING_MULTIPLE_DEFINITIONS EIGEN_UNUSED Packet2d pexp(const Packet2d& x) +{ return pexp_double(x); } + +#endif + } // end namespace internal } // end namespace Eigen -- cgit v1.2.3