![]() |
Eigen
3.3.3
|
00001 // This file is part of Eigen, a lightweight C++ template library 00002 // for linear algebra. 00003 // 00004 // Copyright (C) 2010 Gael Guennebaud <gael.guennebaud@inria.fr> 00005 // 00006 // This Source Code Form is subject to the terms of the Mozilla 00007 // Public License v. 2.0. If a copy of the MPL was not distributed 00008 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/. 00009 00010 #ifndef EIGEN_COMPLEX_SSE_H 00011 #define EIGEN_COMPLEX_SSE_H 00012 00013 namespace Eigen { 00014 00015 namespace internal { 00016 00017 //---------- float ---------- 00018 struct Packet2cf 00019 { 00020 EIGEN_STRONG_INLINE Packet2cf() {} 00021 EIGEN_STRONG_INLINE explicit Packet2cf(const __m128& a) : v(a) {} 00022 __m128 v; 00023 }; 00024 00025 // Use the packet_traits defined in AVX/PacketMath.h instead if we're going 00026 // to leverage AVX instructions. 00027 #ifndef EIGEN_VECTORIZE_AVX 00028 template<> struct packet_traits<std::complex<float> > : default_packet_traits 00029 { 00030 typedef Packet2cf type; 00031 typedef Packet2cf half; 00032 enum { 00033 Vectorizable = 1, 00034 AlignedOnScalar = 1, 00035 size = 2, 00036 HasHalfPacket = 0, 00037 00038 HasAdd = 1, 00039 HasSub = 1, 00040 HasMul = 1, 00041 HasDiv = 1, 00042 HasNegate = 1, 00043 HasAbs = 0, 00044 HasAbs2 = 0, 00045 HasMin = 0, 00046 HasMax = 0, 00047 HasSetLinear = 0, 00048 HasBlend = 1 00049 }; 00050 }; 00051 #endif 00052 00053 template<> struct unpacket_traits<Packet2cf> { typedef std::complex<float> type; enum {size=2, alignment=Aligned16}; typedef Packet2cf half; }; 00054 00055 template<> EIGEN_STRONG_INLINE Packet2cf padd<Packet2cf>(const Packet2cf& a, const Packet2cf& b) { return Packet2cf(_mm_add_ps(a.v,b.v)); } 00056 template<> EIGEN_STRONG_INLINE Packet2cf psub<Packet2cf>(const Packet2cf& a, const Packet2cf& b) { return Packet2cf(_mm_sub_ps(a.v,b.v)); } 00057 template<> EIGEN_STRONG_INLINE Packet2cf pnegate(const Packet2cf& a) 00058 { 00059 const __m128 mask = _mm_castsi128_ps(_mm_setr_epi32(0x80000000,0x80000000,0x80000000,0x80000000)); 00060 return Packet2cf(_mm_xor_ps(a.v,mask)); 00061 } 00062 template<> EIGEN_STRONG_INLINE Packet2cf pconj(const Packet2cf& a) 00063 { 00064 const __m128 mask = _mm_castsi128_ps(_mm_setr_epi32(0x00000000,0x80000000,0x00000000,0x80000000)); 00065 return Packet2cf(_mm_xor_ps(a.v,mask)); 00066 } 00067 00068 template<> EIGEN_STRONG_INLINE Packet2cf pmul<Packet2cf>(const Packet2cf& a, const Packet2cf& b) 00069 { 00070 #ifdef EIGEN_VECTORIZE_SSE3 00071 return Packet2cf(_mm_addsub_ps(_mm_mul_ps(_mm_moveldup_ps(a.v), b.v), 00072 _mm_mul_ps(_mm_movehdup_ps(a.v), 00073 vec4f_swizzle1(b.v, 1, 0, 3, 2)))); 00074 // return Packet2cf(_mm_addsub_ps(_mm_mul_ps(vec4f_swizzle1(a.v, 0, 0, 2, 2), b.v), 00075 // _mm_mul_ps(vec4f_swizzle1(a.v, 1, 1, 3, 3), 00076 // vec4f_swizzle1(b.v, 1, 0, 3, 2)))); 00077 #else 00078 const __m128 mask = _mm_castsi128_ps(_mm_setr_epi32(0x80000000,0x00000000,0x80000000,0x00000000)); 00079 return Packet2cf(_mm_add_ps(_mm_mul_ps(vec4f_swizzle1(a.v, 0, 0, 2, 2), b.v), 00080 _mm_xor_ps(_mm_mul_ps(vec4f_swizzle1(a.v, 1, 1, 3, 3), 00081 vec4f_swizzle1(b.v, 1, 0, 3, 2)), mask))); 00082 #endif 00083 } 00084 00085 template<> EIGEN_STRONG_INLINE Packet2cf pand <Packet2cf>(const Packet2cf& a, const Packet2cf& b) { return Packet2cf(_mm_and_ps(a.v,b.v)); } 00086 template<> EIGEN_STRONG_INLINE Packet2cf por <Packet2cf>(const Packet2cf& a, const Packet2cf& b) { return Packet2cf(_mm_or_ps(a.v,b.v)); } 00087 template<> EIGEN_STRONG_INLINE Packet2cf pxor <Packet2cf>(const Packet2cf& a, const Packet2cf& b) { return Packet2cf(_mm_xor_ps(a.v,b.v)); } 00088 template<> EIGEN_STRONG_INLINE Packet2cf pandnot<Packet2cf>(const Packet2cf& a, const Packet2cf& b) { return Packet2cf(_mm_andnot_ps(a.v,b.v)); } 00089 00090 template<> EIGEN_STRONG_INLINE Packet2cf pload <Packet2cf>(const std::complex<float>* from) { EIGEN_DEBUG_ALIGNED_LOAD return Packet2cf(pload<Packet4f>(&numext::real_ref(*from))); } 00091 template<> EIGEN_STRONG_INLINE Packet2cf ploadu<Packet2cf>(const std::complex<float>* from) { EIGEN_DEBUG_UNALIGNED_LOAD return Packet2cf(ploadu<Packet4f>(&numext::real_ref(*from))); } 00092 00093 template<> EIGEN_STRONG_INLINE Packet2cf pset1<Packet2cf>(const std::complex<float>& from) 00094 { 00095 Packet2cf res; 00096 #if EIGEN_GNUC_AT_MOST(4,2) 00097 // Workaround annoying "may be used uninitialized in this function" warning with gcc 4.2 00098 res.v = _mm_loadl_pi(_mm_set1_ps(0.0f), reinterpret_cast<const __m64*>(&from)); 00099 #elif EIGEN_GNUC_AT_LEAST(4,6) 00100 // Suppress annoying "may be used uninitialized in this function" warning with gcc >= 4.6 00101 #pragma GCC diagnostic push 00102 #pragma GCC diagnostic ignored "-Wuninitialized" 00103 res.v = _mm_loadl_pi(res.v, (const __m64*)&from); 00104 #pragma GCC diagnostic pop 00105 #else 00106 res.v = _mm_loadl_pi(res.v, (const __m64*)&from); 00107 #endif 00108 return Packet2cf(_mm_movelh_ps(res.v,res.v)); 00109 } 00110 00111 template<> EIGEN_STRONG_INLINE Packet2cf ploaddup<Packet2cf>(const std::complex<float>* from) { return pset1<Packet2cf>(*from); } 00112 00113 template<> EIGEN_STRONG_INLINE void pstore <std::complex<float> >(std::complex<float> * to, const Packet2cf& from) { EIGEN_DEBUG_ALIGNED_STORE pstore(&numext::real_ref(*to), Packet4f(from.v)); } 00114 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<float> >(std::complex<float> * to, const Packet2cf& from) { EIGEN_DEBUG_UNALIGNED_STORE pstoreu(&numext::real_ref(*to), Packet4f(from.v)); } 00115 00116 00117 template<> EIGEN_DEVICE_FUNC inline Packet2cf pgather<std::complex<float>, Packet2cf>(const std::complex<float>* from, Index stride) 00118 { 00119 return Packet2cf(_mm_set_ps(std::imag(from[1*stride]), std::real(from[1*stride]), 00120 std::imag(from[0*stride]), std::real(from[0*stride]))); 00121 } 00122 00123 template<> EIGEN_DEVICE_FUNC inline void pscatter<std::complex<float>, Packet2cf>(std::complex<float>* to, const Packet2cf& from, Index stride) 00124 { 00125 to[stride*0] = std::complex<float>(_mm_cvtss_f32(_mm_shuffle_ps(from.v, from.v, 0)), 00126 _mm_cvtss_f32(_mm_shuffle_ps(from.v, from.v, 1))); 00127 to[stride*1] = std::complex<float>(_mm_cvtss_f32(_mm_shuffle_ps(from.v, from.v, 2)), 00128 _mm_cvtss_f32(_mm_shuffle_ps(from.v, from.v, 3))); 00129 } 00130 00131 template<> EIGEN_STRONG_INLINE void prefetch<std::complex<float> >(const std::complex<float> * addr) { _mm_prefetch((const char*)(addr), _MM_HINT_T0); } 00132 00133 template<> EIGEN_STRONG_INLINE std::complex<float> pfirst<Packet2cf>(const Packet2cf& a) 00134 { 00135 #if EIGEN_GNUC_AT_MOST(4,3) 00136 // Workaround gcc 4.2 ICE - this is not performance wise ideal, but who cares... 00137 // This workaround also fix invalid code generation with gcc 4.3 00138 EIGEN_ALIGN16 std::complex<float> res[2]; 00139 _mm_store_ps((float*)res, a.v); 00140 return res[0]; 00141 #else 00142 std::complex<float> res; 00143 _mm_storel_pi((__m64*)&res, a.v); 00144 return res; 00145 #endif 00146 } 00147 00148 template<> EIGEN_STRONG_INLINE Packet2cf preverse(const Packet2cf& a) { return Packet2cf(_mm_castpd_ps(preverse(Packet2d(_mm_castps_pd(a.v))))); } 00149 00150 template<> EIGEN_STRONG_INLINE std::complex<float> predux<Packet2cf>(const Packet2cf& a) 00151 { 00152 return pfirst(Packet2cf(_mm_add_ps(a.v, _mm_movehl_ps(a.v,a.v)))); 00153 } 00154 00155 template<> EIGEN_STRONG_INLINE Packet2cf preduxp<Packet2cf>(const Packet2cf* vecs) 00156 { 00157 return Packet2cf(_mm_add_ps(_mm_movelh_ps(vecs[0].v,vecs[1].v), _mm_movehl_ps(vecs[1].v,vecs[0].v))); 00158 } 00159 00160 template<> EIGEN_STRONG_INLINE std::complex<float> predux_mul<Packet2cf>(const Packet2cf& a) 00161 { 00162 return pfirst(pmul(a, Packet2cf(_mm_movehl_ps(a.v,a.v)))); 00163 } 00164 00165 template<int Offset> 00166 struct palign_impl<Offset,Packet2cf> 00167 { 00168 static EIGEN_STRONG_INLINE void run(Packet2cf& first, const Packet2cf& second) 00169 { 00170 if (Offset==1) 00171 { 00172 first.v = _mm_movehl_ps(first.v, first.v); 00173 first.v = _mm_movelh_ps(first.v, second.v); 00174 } 00175 } 00176 }; 00177 00178 template<> struct conj_helper<Packet2cf, Packet2cf, false,true> 00179 { 00180 EIGEN_STRONG_INLINE Packet2cf pmadd(const Packet2cf& x, const Packet2cf& y, const Packet2cf& c) const 00181 { return padd(pmul(x,y),c); } 00182 00183 EIGEN_STRONG_INLINE Packet2cf pmul(const Packet2cf& a, const Packet2cf& b) const 00184 { 00185 #ifdef EIGEN_VECTORIZE_SSE3 00186 return internal::pmul(a, pconj(b)); 00187 #else 00188 const __m128 mask = _mm_castsi128_ps(_mm_setr_epi32(0x00000000,0x80000000,0x00000000,0x80000000)); 00189 return Packet2cf(_mm_add_ps(_mm_xor_ps(_mm_mul_ps(vec4f_swizzle1(a.v, 0, 0, 2, 2), b.v), mask), 00190 _mm_mul_ps(vec4f_swizzle1(a.v, 1, 1, 3, 3), 00191 vec4f_swizzle1(b.v, 1, 0, 3, 2)))); 00192 #endif 00193 } 00194 }; 00195 00196 template<> struct conj_helper<Packet2cf, Packet2cf, true,false> 00197 { 00198 EIGEN_STRONG_INLINE Packet2cf pmadd(const Packet2cf& x, const Packet2cf& y, const Packet2cf& c) const 00199 { return padd(pmul(x,y),c); } 00200 00201 EIGEN_STRONG_INLINE Packet2cf pmul(const Packet2cf& a, const Packet2cf& b) const 00202 { 00203 #ifdef EIGEN_VECTORIZE_SSE3 00204 return internal::pmul(pconj(a), b); 00205 #else 00206 const __m128 mask = _mm_castsi128_ps(_mm_setr_epi32(0x00000000,0x80000000,0x00000000,0x80000000)); 00207 return Packet2cf(_mm_add_ps(_mm_mul_ps(vec4f_swizzle1(a.v, 0, 0, 2, 2), b.v), 00208 _mm_xor_ps(_mm_mul_ps(vec4f_swizzle1(a.v, 1, 1, 3, 3), 00209 vec4f_swizzle1(b.v, 1, 0, 3, 2)), mask))); 00210 #endif 00211 } 00212 }; 00213 00214 template<> struct conj_helper<Packet2cf, Packet2cf, true,true> 00215 { 00216 EIGEN_STRONG_INLINE Packet2cf pmadd(const Packet2cf& x, const Packet2cf& y, const Packet2cf& c) const 00217 { return padd(pmul(x,y),c); } 00218 00219 EIGEN_STRONG_INLINE Packet2cf pmul(const Packet2cf& a, const Packet2cf& b) const 00220 { 00221 #ifdef EIGEN_VECTORIZE_SSE3 00222 return pconj(internal::pmul(a, b)); 00223 #else 00224 const __m128 mask = _mm_castsi128_ps(_mm_setr_epi32(0x00000000,0x80000000,0x00000000,0x80000000)); 00225 return Packet2cf(_mm_sub_ps(_mm_xor_ps(_mm_mul_ps(vec4f_swizzle1(a.v, 0, 0, 2, 2), b.v), mask), 00226 _mm_mul_ps(vec4f_swizzle1(a.v, 1, 1, 3, 3), 00227 vec4f_swizzle1(b.v, 1, 0, 3, 2)))); 00228 #endif 00229 } 00230 }; 00231 00232 template<> struct conj_helper<Packet4f, Packet2cf, false,false> 00233 { 00234 EIGEN_STRONG_INLINE Packet2cf pmadd(const Packet4f& x, const Packet2cf& y, const Packet2cf& c) const 00235 { return padd(c, pmul(x,y)); } 00236 00237 EIGEN_STRONG_INLINE Packet2cf pmul(const Packet4f& x, const Packet2cf& y) const 00238 { return Packet2cf(Eigen::internal::pmul<Packet4f>(x, y.v)); } 00239 }; 00240 00241 template<> struct conj_helper<Packet2cf, Packet4f, false,false> 00242 { 00243 EIGEN_STRONG_INLINE Packet2cf pmadd(const Packet2cf& x, const Packet4f& y, const Packet2cf& c) const 00244 { return padd(c, pmul(x,y)); } 00245 00246 EIGEN_STRONG_INLINE Packet2cf pmul(const Packet2cf& x, const Packet4f& y) const 00247 { return Packet2cf(Eigen::internal::pmul<Packet4f>(x.v, y)); } 00248 }; 00249 00250 template<> EIGEN_STRONG_INLINE Packet2cf pdiv<Packet2cf>(const Packet2cf& a, const Packet2cf& b) 00251 { 00252 // TODO optimize it for SSE3 and 4 00253 Packet2cf res = conj_helper<Packet2cf,Packet2cf,false,true>().pmul(a,b); 00254 __m128 s = _mm_mul_ps(b.v,b.v); 00255 return Packet2cf(_mm_div_ps(res.v,_mm_add_ps(s,_mm_castsi128_ps(_mm_shuffle_epi32(_mm_castps_si128(s), 0xb1))))); 00256 } 00257 00258 EIGEN_STRONG_INLINE Packet2cf pcplxflip/* <Packet2cf> */(const Packet2cf& x) 00259 { 00260 return Packet2cf(vec4f_swizzle1(x.v, 1, 0, 3, 2)); 00261 } 00262 00263 00264 //---------- double ---------- 00265 struct Packet1cd 00266 { 00267 EIGEN_STRONG_INLINE Packet1cd() {} 00268 EIGEN_STRONG_INLINE explicit Packet1cd(const __m128d& a) : v(a) {} 00269 __m128d v; 00270 }; 00271 00272 // Use the packet_traits defined in AVX/PacketMath.h instead if we're going 00273 // to leverage AVX instructions. 00274 #ifndef EIGEN_VECTORIZE_AVX 00275 template<> struct packet_traits<std::complex<double> > : default_packet_traits 00276 { 00277 typedef Packet1cd type; 00278 typedef Packet1cd half; 00279 enum { 00280 Vectorizable = 1, 00281 AlignedOnScalar = 0, 00282 size = 1, 00283 HasHalfPacket = 0, 00284 00285 HasAdd = 1, 00286 HasSub = 1, 00287 HasMul = 1, 00288 HasDiv = 1, 00289 HasNegate = 1, 00290 HasAbs = 0, 00291 HasAbs2 = 0, 00292 HasMin = 0, 00293 HasMax = 0, 00294 HasSetLinear = 0 00295 }; 00296 }; 00297 #endif 00298 00299 template<> struct unpacket_traits<Packet1cd> { typedef std::complex<double> type; enum {size=1, alignment=Aligned16}; typedef Packet1cd half; }; 00300 00301 template<> EIGEN_STRONG_INLINE Packet1cd padd<Packet1cd>(const Packet1cd& a, const Packet1cd& b) { return Packet1cd(_mm_add_pd(a.v,b.v)); } 00302 template<> EIGEN_STRONG_INLINE Packet1cd psub<Packet1cd>(const Packet1cd& a, const Packet1cd& b) { return Packet1cd(_mm_sub_pd(a.v,b.v)); } 00303 template<> EIGEN_STRONG_INLINE Packet1cd pnegate(const Packet1cd& a) { return Packet1cd(pnegate(Packet2d(a.v))); } 00304 template<> EIGEN_STRONG_INLINE Packet1cd pconj(const Packet1cd& a) 00305 { 00306 const __m128d mask = _mm_castsi128_pd(_mm_set_epi32(0x80000000,0x0,0x0,0x0)); 00307 return Packet1cd(_mm_xor_pd(a.v,mask)); 00308 } 00309 00310 template<> EIGEN_STRONG_INLINE Packet1cd pmul<Packet1cd>(const Packet1cd& a, const Packet1cd& b) 00311 { 00312 #ifdef EIGEN_VECTORIZE_SSE3 00313 return Packet1cd(_mm_addsub_pd(_mm_mul_pd(_mm_movedup_pd(a.v), b.v), 00314 _mm_mul_pd(vec2d_swizzle1(a.v, 1, 1), 00315 vec2d_swizzle1(b.v, 1, 0)))); 00316 #else 00317 const __m128d mask = _mm_castsi128_pd(_mm_set_epi32(0x0,0x0,0x80000000,0x0)); 00318 return Packet1cd(_mm_add_pd(_mm_mul_pd(vec2d_swizzle1(a.v, 0, 0), b.v), 00319 _mm_xor_pd(_mm_mul_pd(vec2d_swizzle1(a.v, 1, 1), 00320 vec2d_swizzle1(b.v, 1, 0)), mask))); 00321 #endif 00322 } 00323 00324 template<> EIGEN_STRONG_INLINE Packet1cd pand <Packet1cd>(const Packet1cd& a, const Packet1cd& b) { return Packet1cd(_mm_and_pd(a.v,b.v)); } 00325 template<> EIGEN_STRONG_INLINE Packet1cd por <Packet1cd>(const Packet1cd& a, const Packet1cd& b) { return Packet1cd(_mm_or_pd(a.v,b.v)); } 00326 template<> EIGEN_STRONG_INLINE Packet1cd pxor <Packet1cd>(const Packet1cd& a, const Packet1cd& b) { return Packet1cd(_mm_xor_pd(a.v,b.v)); } 00327 template<> EIGEN_STRONG_INLINE Packet1cd pandnot<Packet1cd>(const Packet1cd& a, const Packet1cd& b) { return Packet1cd(_mm_andnot_pd(a.v,b.v)); } 00328 00329 // FIXME force unaligned load, this is a temporary fix 00330 template<> EIGEN_STRONG_INLINE Packet1cd pload <Packet1cd>(const std::complex<double>* from) 00331 { EIGEN_DEBUG_ALIGNED_LOAD return Packet1cd(pload<Packet2d>((const double*)from)); } 00332 template<> EIGEN_STRONG_INLINE Packet1cd ploadu<Packet1cd>(const std::complex<double>* from) 00333 { EIGEN_DEBUG_UNALIGNED_LOAD return Packet1cd(ploadu<Packet2d>((const double*)from)); } 00334 template<> EIGEN_STRONG_INLINE Packet1cd pset1<Packet1cd>(const std::complex<double>& from) 00335 { /* here we really have to use unaligned loads :( */ return ploadu<Packet1cd>(&from); } 00336 00337 template<> EIGEN_STRONG_INLINE Packet1cd ploaddup<Packet1cd>(const std::complex<double>* from) { return pset1<Packet1cd>(*from); } 00338 00339 // FIXME force unaligned store, this is a temporary fix 00340 template<> EIGEN_STRONG_INLINE void pstore <std::complex<double> >(std::complex<double> * to, const Packet1cd& from) { EIGEN_DEBUG_ALIGNED_STORE pstore((double*)to, Packet2d(from.v)); } 00341 template<> EIGEN_STRONG_INLINE void pstoreu<std::complex<double> >(std::complex<double> * to, const Packet1cd& from) { EIGEN_DEBUG_UNALIGNED_STORE pstoreu((double*)to, Packet2d(from.v)); } 00342 00343 template<> EIGEN_STRONG_INLINE void prefetch<std::complex<double> >(const std::complex<double> * addr) { _mm_prefetch((const char*)(addr), _MM_HINT_T0); } 00344 00345 template<> EIGEN_STRONG_INLINE std::complex<double> pfirst<Packet1cd>(const Packet1cd& a) 00346 { 00347 EIGEN_ALIGN16 double res[2]; 00348 _mm_store_pd(res, a.v); 00349 return std::complex<double>(res[0],res[1]); 00350 } 00351 00352 template<> EIGEN_STRONG_INLINE Packet1cd preverse(const Packet1cd& a) { return a; } 00353 00354 template<> EIGEN_STRONG_INLINE std::complex<double> predux<Packet1cd>(const Packet1cd& a) 00355 { 00356 return pfirst(a); 00357 } 00358 00359 template<> EIGEN_STRONG_INLINE Packet1cd preduxp<Packet1cd>(const Packet1cd* vecs) 00360 { 00361 return vecs[0]; 00362 } 00363 00364 template<> EIGEN_STRONG_INLINE std::complex<double> predux_mul<Packet1cd>(const Packet1cd& a) 00365 { 00366 return pfirst(a); 00367 } 00368 00369 template<int Offset> 00370 struct palign_impl<Offset,Packet1cd> 00371 { 00372 static EIGEN_STRONG_INLINE void run(Packet1cd& /*first*/, const Packet1cd& /*second*/) 00373 { 00374 // FIXME is it sure we never have to align a Packet1cd? 00375 // Even though a std::complex<double> has 16 bytes, it is not necessarily aligned on a 16 bytes boundary... 00376 } 00377 }; 00378 00379 template<> struct conj_helper<Packet1cd, Packet1cd, false,true> 00380 { 00381 EIGEN_STRONG_INLINE Packet1cd pmadd(const Packet1cd& x, const Packet1cd& y, const Packet1cd& c) const 00382 { return padd(pmul(x,y),c); } 00383 00384 EIGEN_STRONG_INLINE Packet1cd pmul(const Packet1cd& a, const Packet1cd& b) const 00385 { 00386 #ifdef EIGEN_VECTORIZE_SSE3 00387 return internal::pmul(a, pconj(b)); 00388 #else 00389 const __m128d mask = _mm_castsi128_pd(_mm_set_epi32(0x80000000,0x0,0x0,0x0)); 00390 return Packet1cd(_mm_add_pd(_mm_xor_pd(_mm_mul_pd(vec2d_swizzle1(a.v, 0, 0), b.v), mask), 00391 _mm_mul_pd(vec2d_swizzle1(a.v, 1, 1), 00392 vec2d_swizzle1(b.v, 1, 0)))); 00393 #endif 00394 } 00395 }; 00396 00397 template<> struct conj_helper<Packet1cd, Packet1cd, true,false> 00398 { 00399 EIGEN_STRONG_INLINE Packet1cd pmadd(const Packet1cd& x, const Packet1cd& y, const Packet1cd& c) const 00400 { return padd(pmul(x,y),c); } 00401 00402 EIGEN_STRONG_INLINE Packet1cd pmul(const Packet1cd& a, const Packet1cd& b) const 00403 { 00404 #ifdef EIGEN_VECTORIZE_SSE3 00405 return internal::pmul(pconj(a), b); 00406 #else 00407 const __m128d mask = _mm_castsi128_pd(_mm_set_epi32(0x80000000,0x0,0x0,0x0)); 00408 return Packet1cd(_mm_add_pd(_mm_mul_pd(vec2d_swizzle1(a.v, 0, 0), b.v), 00409 _mm_xor_pd(_mm_mul_pd(vec2d_swizzle1(a.v, 1, 1), 00410 vec2d_swizzle1(b.v, 1, 0)), mask))); 00411 #endif 00412 } 00413 }; 00414 00415 template<> struct conj_helper<Packet1cd, Packet1cd, true,true> 00416 { 00417 EIGEN_STRONG_INLINE Packet1cd pmadd(const Packet1cd& x, const Packet1cd& y, const Packet1cd& c) const 00418 { return padd(pmul(x,y),c); } 00419 00420 EIGEN_STRONG_INLINE Packet1cd pmul(const Packet1cd& a, const Packet1cd& b) const 00421 { 00422 #ifdef EIGEN_VECTORIZE_SSE3 00423 return pconj(internal::pmul(a, b)); 00424 #else 00425 const __m128d mask = _mm_castsi128_pd(_mm_set_epi32(0x80000000,0x0,0x0,0x0)); 00426 return Packet1cd(_mm_sub_pd(_mm_xor_pd(_mm_mul_pd(vec2d_swizzle1(a.v, 0, 0), b.v), mask), 00427 _mm_mul_pd(vec2d_swizzle1(a.v, 1, 1), 00428 vec2d_swizzle1(b.v, 1, 0)))); 00429 #endif 00430 } 00431 }; 00432 00433 template<> struct conj_helper<Packet2d, Packet1cd, false,false> 00434 { 00435 EIGEN_STRONG_INLINE Packet1cd pmadd(const Packet2d& x, const Packet1cd& y, const Packet1cd& c) const 00436 { return padd(c, pmul(x,y)); } 00437 00438 EIGEN_STRONG_INLINE Packet1cd pmul(const Packet2d& x, const Packet1cd& y) const 00439 { return Packet1cd(Eigen::internal::pmul<Packet2d>(x, y.v)); } 00440 }; 00441 00442 template<> struct conj_helper<Packet1cd, Packet2d, false,false> 00443 { 00444 EIGEN_STRONG_INLINE Packet1cd pmadd(const Packet1cd& x, const Packet2d& y, const Packet1cd& c) const 00445 { return padd(c, pmul(x,y)); } 00446 00447 EIGEN_STRONG_INLINE Packet1cd pmul(const Packet1cd& x, const Packet2d& y) const 00448 { return Packet1cd(Eigen::internal::pmul<Packet2d>(x.v, y)); } 00449 }; 00450 00451 template<> EIGEN_STRONG_INLINE Packet1cd pdiv<Packet1cd>(const Packet1cd& a, const Packet1cd& b) 00452 { 00453 // TODO optimize it for SSE3 and 4 00454 Packet1cd res = conj_helper<Packet1cd,Packet1cd,false,true>().pmul(a,b); 00455 __m128d s = _mm_mul_pd(b.v,b.v); 00456 return Packet1cd(_mm_div_pd(res.v, _mm_add_pd(s,_mm_shuffle_pd(s, s, 0x1)))); 00457 } 00458 00459 EIGEN_STRONG_INLINE Packet1cd pcplxflip/* <Packet1cd> */(const Packet1cd& x) 00460 { 00461 return Packet1cd(preverse(Packet2d(x.v))); 00462 } 00463 00464 EIGEN_DEVICE_FUNC inline void 00465 ptranspose(PacketBlock<Packet2cf,2>& kernel) { 00466 __m128d w1 = _mm_castps_pd(kernel.packet[0].v); 00467 __m128d w2 = _mm_castps_pd(kernel.packet[1].v); 00468 00469 __m128 tmp = _mm_castpd_ps(_mm_unpackhi_pd(w1, w2)); 00470 kernel.packet[0].v = _mm_castpd_ps(_mm_unpacklo_pd(w1, w2)); 00471 kernel.packet[1].v = tmp; 00472 } 00473 00474 template<> EIGEN_STRONG_INLINE Packet2cf pblend(const Selector<2>& ifPacket, const Packet2cf& thenPacket, const Packet2cf& elsePacket) { 00475 __m128d result = pblend<Packet2d>(ifPacket, _mm_castps_pd(thenPacket.v), _mm_castps_pd(elsePacket.v)); 00476 return Packet2cf(_mm_castpd_ps(result)); 00477 } 00478 00479 template<> EIGEN_STRONG_INLINE Packet2cf pinsertfirst(const Packet2cf& a, std::complex<float> b) 00480 { 00481 return Packet2cf(_mm_loadl_pi(a.v, reinterpret_cast<const __m64*>(&b))); 00482 } 00483 00484 template<> EIGEN_STRONG_INLINE Packet1cd pinsertfirst(const Packet1cd&, std::complex<double> b) 00485 { 00486 return pset1<Packet1cd>(b); 00487 } 00488 00489 template<> EIGEN_STRONG_INLINE Packet2cf pinsertlast(const Packet2cf& a, std::complex<float> b) 00490 { 00491 return Packet2cf(_mm_loadh_pi(a.v, reinterpret_cast<const __m64*>(&b))); 00492 } 00493 00494 template<> EIGEN_STRONG_INLINE Packet1cd pinsertlast(const Packet1cd&, std::complex<double> b) 00495 { 00496 return pset1<Packet1cd>(b); 00497 } 00498 00499 } // end namespace internal 00500 00501 } // end namespace Eigen 00502 00503 #endif // EIGEN_COMPLEX_SSE_H