portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2024-11-10 07:55:35 +00:00
Code Releases Activity

ExtractSlice / InsertSlice for lower dimensional lattices where the lattice is not

Browse Source 
distributed in the orthogonal direction.
Useful for fermion 4d/5d etc..
This commit is contained in:
paboyle 2016-05-11 14:12:02 +01:00
parent f8f9fd6f22
commit aceaee774c
1 changed files with 84 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
lib/lattice/Lattice_transfer.h
View File

@ -325,6 +325,89 @@ inline void blockPromote(const Lattice<iVector<CComplex,nbasis > > &coarseData,
}
template<class vobj>
void InsertSlice(const Lattice<vobj> &lowDim,Lattice<vobj> & higherDim,int slice, int orthog)
{
typedef typename vobj::scalar_object sobj;
sobj s;
GridBase *lg = lowdim._grid;
GridBase *hg = higherdim._grid;
int nl = lg->_ndimension;
int nh = hg->_ndimension;
assert(nl+1 == nh);
assert(orthog<nh);
assert(orthog>=0);
assert(hg->_processors[orthog]==0);
int dl; dl = 0;
for(int d=0;d<nh;d++){
if ( d != orthog) {
assert(lg->_processors[dl] == hg->_processors[d]);
assert(lg->_ldimensions[dl] == hg->_ldimensions[d]);
dl++;
}
}
// the above should guarantee that the operations are local
std::vector<int> lcoor(nl);
std::vector<int> hcoor(nh);
hcoor[orthog] = slice;
for(int idx=0;idx<lg->lSites();idx++){
lg->LocalIndexToLocalCoor(idx,lcoor);
dl=0;
for(int d=0;d<nh;d++){
if ( d!orthog ) {
hcoor[d]=lcoor[ld++];
}
}
peekLocalSite(s,higherDim,hcoor);
pokeLocalSite(s,lowDim,lcoor);
}
}
template<class vobj>
void ExtractSlice(Lattice<vobj> &lowDim,const Lattice<vobj> & higherDim,int slice, int dir)
{
typedef typename vobj::scalar_object sobj;
sobj s;
GridBase *lg = lowdim._grid;
GridBase *hg = higherdim._grid;
int nl = lg->_ndimension;
int nh = hg->_ndimension;
assert(nl+1 == nh);
assert(orthog<nh);
assert(orthog>=0);
assert(hg->_processors[orthog]==0);
int dl; dl = 0;
for(int d=0;d<nh;d++){
if ( d != orthog) {
assert(lg->_processors[dl] == hg->_processors[d]);
assert(lg->_ldimensions[dl] == hg->_ldimensions[d]);
dl++;
}
}
// the above should guarantee that the operations are local
std::vector<int> lcoor(nl);
std::vector<int> hcoor(nh);
hcoor[orthog] = slice;
for(int idx=0;idx<lg->lSites();idx++){
lg->LocalIndexToLocalCoor(idx,lcoor);
dl=0;
for(int d=0;d<nh;d++){
if ( d!orthog ) {
hcoor[d]=lcoor[ld++];
}
}
peekLocalSite(s,lowDim,lcoor);
pokeLocalSite(s,higherDim,hcoor);
}
}
template<class vobj>
void Replicate(Lattice<vobj> &coarse,Lattice<vobj> & fine)