From c7eb3a74cb4d1d2e3d2db5b5c69c1ab27f47d76b Mon Sep 17 00:00:00 2001 From: David Tellenbach Date: Fri, 11 Dec 2020 12:41:52 +0100 Subject: Don't guard psqrt for std::complex with EIGEN_ARCH_ARM64 --- Eigen/src/Core/arch/NEON/Complex.h | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'Eigen/src/Core') diff --git a/Eigen/src/Core/arch/NEON/Complex.h b/Eigen/src/Core/arch/NEON/Complex.h index 2210d5a78..a889ab1d2 100644 --- a/Eigen/src/Core/arch/NEON/Complex.h +++ b/Eigen/src/Core/arch/NEON/Complex.h @@ -432,6 +432,14 @@ EIGEN_DEVICE_FUNC inline void ptranspose(PacketBlock& kernel) kernel.packet[1].v = tmp; } +template<> EIGEN_STRONG_INLINE Packet1cf psqrt(const Packet1cf& a) { + return psqrt_complex(a); +} + +template<> EIGEN_STRONG_INLINE Packet2cf psqrt(const Packet2cf& a) { + return psqrt_complex(a); +} + //---------- double ---------- #if EIGEN_ARCH_ARM64 && !EIGEN_APPLE_DOUBLE_NEON_BUG @@ -644,14 +652,6 @@ EIGEN_STRONG_INLINE void ptranspose(PacketBlock& kernel) kernel.packet[1].v = tmp; } -template<> EIGEN_STRONG_INLINE Packet1cf psqrt(const Packet1cf& a) { - return psqrt_complex(a); -} - -template<> EIGEN_STRONG_INLINE Packet2cf psqrt(const Packet2cf& a) { - return psqrt_complex(a); -} - template<> EIGEN_STRONG_INLINE Packet1cd psqrt(const Packet1cd& a) { return psqrt_complex(a); } -- cgit v1.2.3