diff options
| author |  Joel Holdsworth <joel.holdsworth@vcatechnology.com> | 2019-12-17 18:41:16 +0000 |
|---|---|---|
| committer | Joel Holdsworth <joel.holdsworth@vcatechnology.com> | 2020-03-10 22:46:19 +0000 |
| commit | d5d3cf9339144ddfc4392af5a7685ee4d0a71a4d (patch) | |
| tree | 068439ed744ed38852077060b21440ad636f8396 /Eigen/src/Core/arch/NEON/TypeCasting.h | |
| parent | eacf97f72749da0ed29b76a07ad3f23c1ec81a4a (diff) | |
NEON: Added uint32_t packet math
Diffstat (limited to 'Eigen/src/Core/arch/NEON/TypeCasting.h')
| -rw-r--r-- | Eigen/src/Core/arch/NEON/TypeCasting.h | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/Eigen/src/Core/arch/NEON/TypeCasting.h b/Eigen/src/Core/arch/NEON/TypeCasting.h index dfd6623cc..298088c2d 100644 --- a/Eigen/src/Core/arch/NEON/TypeCasting.h +++ b/Eigen/src/Core/arch/NEON/TypeCasting.h @@ -16,22 +16,38 @@ namespace internal { template<> struct type_casting_traits<float,numext::int32_t> { enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 }; }; +template<> struct type_casting_traits<float,numext::uint32_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<> struct type_casting_traits<numext::uint32_t,float> +{ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 }; }; template<> EIGEN_STRONG_INLINE Packet2f pcast<Packet2i,Packet2f>(const Packet2i& a) { return vcvt_f32_s32(a); } +template<> EIGEN_STRONG_INLINE Packet2f pcast<Packet2ui,Packet2f>(const Packet2ui& a) { return vcvt_f32_u32(a); } template<> EIGEN_STRONG_INLINE Packet4f pcast<Packet4i,Packet4f>(const Packet4i& a) { return vcvtq_f32_s32(a); } +template<> EIGEN_STRONG_INLINE Packet4f pcast<Packet4ui,Packet4f>(const Packet4ui& a) { return vcvtq_f32_u32(a); } template<> EIGEN_STRONG_INLINE Packet2i pcast<Packet2f,Packet2i>(const Packet2f& a) { return vcvt_s32_f32(a); } template<> EIGEN_STRONG_INLINE Packet4i pcast<Packet4f,Packet4i>(const Packet4f& a) { return vcvtq_s32_f32(a); } +template<> EIGEN_STRONG_INLINE Packet2ui pcast<Packet2f,Packet2ui>(const Packet2f& a) { return vcvt_u32_f32(a); } +template<> EIGEN_STRONG_INLINE Packet4ui pcast<Packet4f,Packet4ui>(const Packet4f& a) { return vcvtq_u32_f32(a); } template<> EIGEN_STRONG_INLINE Packet2f preinterpret<Packet2f,Packet2i>(const Packet2i& a) { return vreinterpret_f32_s32(a); } +template<> EIGEN_STRONG_INLINE Packet2f preinterpret<Packet2f,Packet2ui>(const Packet2ui& a) +{ return vreinterpret_f32_u32(a); } template<> EIGEN_STRONG_INLINE Packet4f preinterpret<Packet4f,Packet4i>(const Packet4i& a) { return vreinterpretq_f32_s32(a); } +template<> EIGEN_STRONG_INLINE Packet4f preinterpret<Packet4f,Packet4ui>(const Packet4ui& a) +{ return vreinterpretq_f32_u32(a); } template<> EIGEN_STRONG_INLINE Packet2i preinterpret<Packet2i,Packet2f>(const Packet2f& a) { return vreinterpret_s32_f32(a); } template<> EIGEN_STRONG_INLINE Packet4i preinterpret<Packet4i,Packet4f>(const Packet4f& a) { return vreinterpretq_s32_f32(a); } +template<> EIGEN_STRONG_INLINE Packet2ui preinterpret<Packet2ui,Packet2f>(const Packet2f& a) +{ return vreinterpret_u32_f32(a); } +template<> EIGEN_STRONG_INLINE Packet4ui preinterpret<Packet4ui,Packet4f>(const Packet4f& a) +{ return vreinterpretq_u32_f32(a); } } // end namespace internal |