diff options
| author |  Ilya Tokar <tokarip@google.com> | 2019-12-16 16:00:35 -0500 |
|---|---|---|
| committer | Ilya Tokar <tokarip@google.com> | 2020-01-07 21:22:44 +0000 |
| commit | 19876ced76bd1730008e02fc4c43c2228faed38a (patch) | |
| tree | 22f999134e79f84f6f9dc0d19bdfa6740db416f0 /Eigen/src/Core/arch/SSE/PacketMath.h | |
| parent | d0ae052da4ce25a5b4306bfbb5bf8edcd010b663 (diff) | |
Bug #1785: Introduce numext::rint.
This provides a new op that matches std::rint and previous behavior of
pround. Also adds corresponding unsupported/../Tensor op.
Performance is the same as e. g. floor (tested SSE/AVX).
Diffstat (limited to 'Eigen/src/Core/arch/SSE/PacketMath.h')
| -rwxr-xr-x | Eigen/src/Core/arch/SSE/PacketMath.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/Eigen/src/Core/arch/SSE/PacketMath.h b/Eigen/src/Core/arch/SSE/PacketMath.h index 2f50326cb..d6a4a5c7f 100755 --- a/Eigen/src/Core/arch/SSE/PacketMath.h +++ b/Eigen/src/Core/arch/SSE/PacketMath.h @@ -124,6 +124,7 @@ struct packet_traits<float> : default_packet_traits { #ifdef EIGEN_VECTORIZE_SSE4_1 , + HasRint = 1, HasRound = 1, HasCeil = 1 #endif @@ -148,6 +149,7 @@ struct packet_traits<double> : default_packet_traits { #ifdef EIGEN_VECTORIZE_SSE4_1 , HasRound = 1, + HasRint = 1, HasFloor = 1, HasCeil = 1 #endif @@ -443,6 +445,9 @@ template<> EIGEN_STRONG_INLINE Packet2d pround<Packet2d>(const Packet2d& a) return _mm_round_pd(padd(por(pand(a, mask), prev0dot5), a), _MM_FROUND_TO_ZERO); } +template<> EIGEN_STRONG_INLINE Packet4f print<Packet4f>(const Packet4f& a) { return _mm_round_ps(a, _MM_FROUND_CUR_DIRECTION); } +template<> EIGEN_STRONG_INLINE Packet2d print<Packet2d>(const Packet2d& a) { return _mm_round_pd(a, _MM_FROUND_CUR_DIRECTION); } + template<> EIGEN_STRONG_INLINE Packet4f pceil<Packet4f>(const Packet4f& a) { return _mm_ceil_ps(a); } template<> EIGEN_STRONG_INLINE Packet2d pceil<Packet2d>(const Packet2d& a) { return _mm_ceil_pd(a); } |