diff options
| author |  Joel Holdsworth <joel@airwebreathe.org.uk> | 2020-01-10 18:31:15 +0000 |
|---|---|---|
| committer |  Rasmus Munk Larsen <rmlarsen@google.com> | 2020-01-10 18:31:15 +0000 |
| commit | da5a7afed056596b089a4241b62a7e17f2c43119 (patch) | |
| tree | 037732af247a28a1598c3d6e4ddcbffd2e7404be /Eigen/src/Core/arch/NEON/TypeCasting.h | |
| parent | 452371cead5e83131a816b1069e1aec4b9c6cf36 (diff) | |
Improvements to the tidiness and completeness of the NEON implementation
Diffstat (limited to 'Eigen/src/Core/arch/NEON/TypeCasting.h')
| -rw-r--r-- | Eigen/src/Core/arch/NEON/TypeCasting.h | 48 |
1 files changed, 14 insertions, 34 deletions
diff --git a/Eigen/src/Core/arch/NEON/TypeCasting.h b/Eigen/src/Core/arch/NEON/TypeCasting.h index 20dbe1332..5036ee92b 100644 --- a/Eigen/src/Core/arch/NEON/TypeCasting.h +++ b/Eigen/src/Core/arch/NEON/TypeCasting.h @@ -10,44 +10,24 @@ #ifndef EIGEN_TYPE_CASTING_NEON_H #define EIGEN_TYPE_CASTING_NEON_H +#include <Eigen/src/Core/util/Meta.h> + namespace Eigen { namespace internal { -template <> -struct type_casting_traits<float, int> { - enum { - VectorizedCast = 1, - SrcCoeffRatio = 1, - TgtCoeffRatio = 1 - }; -}; - -template <> -struct type_casting_traits<int, float> { - enum { - VectorizedCast = 1, - SrcCoeffRatio = 1, - TgtCoeffRatio = 1 - }; -}; - - -template<> EIGEN_STRONG_INLINE Packet4i pcast<Packet4f, Packet4i>(const Packet4f& a) { - return vcvtq_s32_f32(a); -} - -template<> EIGEN_STRONG_INLINE Packet4f pcast<Packet4i, Packet4f>(const Packet4i& a) { - return vcvtq_f32_s32(a); -} - -template<> EIGEN_STRONG_INLINE Packet4i preinterpret<Packet4i,Packet4f>(const Packet4f& a) { - return vreinterpretq_s32_f32(a); -} - -template<> EIGEN_STRONG_INLINE Packet4f preinterpret<Packet4f,Packet4i>(const Packet4i& a) { - return vreinterpretq_f32_s32(a); -} +template<> struct type_casting_traits<float,numext::int32_t> +{ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 }; }; +template<> struct type_casting_traits<numext::int32_t,float> +{ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 }; }; + +template<> EIGEN_STRONG_INLINE Packet4f pcast<Packet4i,Packet4f>(const Packet4i& a) { return vcvtq_f32_s32(a); } +template<> EIGEN_STRONG_INLINE Packet4i pcast<Packet4f,Packet4i>(const Packet4f& a) { return vcvtq_s32_f32(a); } + +template<> EIGEN_STRONG_INLINE Packet4f preinterpret<Packet4f,Packet4i>(const Packet4i& a) +{ return vreinterpretq_f32_s32(a); } +template<> EIGEN_STRONG_INLINE Packet4i preinterpret<Packet4i,Packet4f>(const Packet4f& a) +{ return vreinterpretq_s32_f32(a); } } // end namespace internal |