portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-09 21:50:45 +01:00
Code Releases Activity

Rotation needed for hopping term in fifth dim with Ls vectorised fields

Browse Source
This commit is contained in:
paboyle 2016-07-14 23:45:36 +01:00
parent 42af132dab
commit 8f47d0b5ab
1 changed files with 29 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
lib/simd/Grid_vector_types.h
View File

@ -303,13 +303,14 @@ namespace Grid {
int dist = perm&0xF; int dist = perm&0xF;
y=rotate(b,dist); y=rotate(b,dist);
return; return;
} } else {
switch(perm){ switch(perm){
case 3: permute3(y,b); break; case 3: permute3(y,b); break;
case 2: permute2(y,b); break; case 2: permute2(y,b); break;
case 1: permute1(y,b); break; case 1: permute1(y,b); break;
case 0: permute0(y,b); break; case 0: permute0(y,b); break;
default: assert(0); default: assert(0);
}
} }
} }
@ -336,6 +337,20 @@ namespace Grid {
ret.v = Optimization::Rotate::rotate(b.v,2*nrot); ret.v = Optimization::Rotate::rotate(b.v,2*nrot);
return ret; return ret;
} }
template <class S, class V, IfNotComplex<S> =0>
inline void rotate( Grid_simd<S,V> &ret,Grid_simd<S,V> b,int nrot)
{
nrot = nrot % Grid_simd<S,V>::Nsimd();
// std::cout << "Rotate Real by "<<nrot<<std::endl;
ret.v = Optimization::Rotate::rotate(b.v,nrot);
}
template <class S, class V, IfComplex<S> =0>
inline void rotate(Grid_simd<S,V> &ret,Grid_simd<S,V> b,int nrot)
{
nrot = nrot % Grid_simd<S,V>::Nsimd();
// std::cout << "Rotate Complex by "<<nrot<<std::endl;
ret.v = Optimization::Rotate::rotate(b.v,2*nrot);
}
/////////////////////// ///////////////////////
// Splat // Splat
@ -347,6 +362,12 @@ namespace Grid {
ret.v = binary<V>(a, b, VsplatSIMD()); ret.v = binary<V>(a, b, VsplatSIMD());
} }
template <class S, class V>
inline void vbroadcast(Grid_simd<S,V> &ret,const Grid_simd<S,V> &src,int lane){
S* typepun =(S*) &src;
vsplat(ret,typepun[lane]);
}
// overload if complex // overload if complex
template <class S,class V> inline void vsplat(Grid_simd<S,V> &ret, EnableIf<is_complex < S >, S> c) { template <class S,class V> inline void vsplat(Grid_simd<S,V> &ret, EnableIf<is_complex < S >, S> c) {
vsplat(ret,real(c),imag(c)); vsplat(ret,real(c),imag(c));
@ -354,7 +375,7 @@ namespace Grid {
//if real fill with a, if complex fill with a in the real part (first function above) //if real fill with a, if complex fill with a in the real part (first function above)
template <class S,class V> template <class S,class V>
inline void vsplat(Grid_simd<S,V> &ret,NotEnableIf<is_complex< S>,S> a){ inline void vsplat(Grid_simd<S,V> &ret,NotEnableIf<is_complex< S>,S> a){
ret.v = unary<V>(a, VsplatSIMD()); ret.v = unary<V>(a, VsplatSIMD());
} }
////////////////////////// //////////////////////////