1 // This file is part of Eigen, a lightweight C++ template library 2 // for linear algebra. 3 // 4 // Copyright (C) 2015 Benoit Steiner <benoit.steiner.goog (at) gmail.com> 5 // 6 // This Source Code Form is subject to the terms of the Mozilla 7 // Public License v. 2.0. If a copy of the MPL was not distributed 8 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/. 9 10 #ifndef EIGEN_TYPE_CASTING_SSE_H 11 #define EIGEN_TYPE_CASTING_SSE_H 12 13 namespace Eigen { 14 15 namespace internal { 16 17 template <> 18 struct type_casting_traits<float, int> { 19 enum { 20 VectorizedCast = 1, 21 SrcCoeffRatio = 1, 22 TgtCoeffRatio = 1 23 }; 24 }; 25 26 template<> EIGEN_STRONG_INLINE Packet4i pcast<Packet4f, Packet4i>(const Packet4f& a) { 27 return _mm_cvttps_epi32(a); 28 } 29 30 31 template <> 32 struct type_casting_traits<int, float> { 33 enum { 34 VectorizedCast = 1, 35 SrcCoeffRatio = 1, 36 TgtCoeffRatio = 1 37 }; 38 }; 39 40 template<> EIGEN_STRONG_INLINE Packet4f pcast<Packet4i, Packet4f>(const Packet4i& a) { 41 return _mm_cvtepi32_ps(a); 42 } 43 44 45 template <> 46 struct type_casting_traits<double, float> { 47 enum { 48 VectorizedCast = 1, 49 SrcCoeffRatio = 2, 50 TgtCoeffRatio = 1 51 }; 52 }; 53 54 template<> EIGEN_STRONG_INLINE Packet4f pcast<Packet2d, Packet4f>(const Packet2d& a, const Packet2d& b) { 55 return _mm_shuffle_ps(_mm_cvtpd_ps(a), _mm_cvtpd_ps(b), (1 << 2) | (1 << 6)); 56 } 57 58 template <> 59 struct type_casting_traits<float, double> { 60 enum { 61 VectorizedCast = 1, 62 SrcCoeffRatio = 1, 63 TgtCoeffRatio = 2 64 }; 65 }; 66 67 template<> EIGEN_STRONG_INLINE Packet2d pcast<Packet4f, Packet2d>(const Packet4f& a) { 68 // Simply discard the second half of the input 69 return _mm_cvtps_pd(a); 70 } 71 72 73 } // end namespace internal 74 75 } // end namespace Eigen 76 77 #endif // EIGEN_TYPE_CASTING_SSE_H 78
