portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-04 19:25:56 +01:00
Code Releases Activity

Threaded the constructor of GeneralLocalStencil

Browse Source
This commit is contained in:
Christopher Kelly 2023-06-23 09:57:38 -04:00
parent 4241c7d4a3
commit 36cc9c524f
1 changed files with 52 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
Grid/stencil/GeneralLocalStencil.h
View File

@ -79,60 +79,60 @@ public:
this->_entries.resize(npoints* osites); this->_entries.resize(npoints* osites);
this->_entries_p = &_entries[0]; this->_entries_p = &_entries[0];
thread_for(site, osites, {
Coordinate Coor;
Coordinate NbrCoor;
Coordinate Coor; for(Integer ii=0;ii<npoints;ii++){
Coordinate NbrCoor; Integer lex = site*npoints+ii;
for(Integer site=0;site<osites;site++){ GeneralStencilEntry SE;
for(Integer ii=0;ii<npoints;ii++){ ////////////////////////////////////////////////
Integer lex = site*npoints+ii; // Outer index of neighbour Offset calculation
GeneralStencilEntry SE; ////////////////////////////////////////////////
//////////////////////////////////////////////// grid->oCoorFromOindex(Coor,site);
// Outer index of neighbour Offset calculation for(int d=0;d<Coor.size();d++){
//////////////////////////////////////////////// int rd = grid->_rdimensions[d];
grid->oCoorFromOindex(Coor,site); NbrCoor[d] = (Coor[d] + shifts[ii][d] + rd )%rd;
for(int d=0;d<Coor.size();d++){ }
int rd = grid->_rdimensions[d]; SE._offset = grid->oIndexReduced(NbrCoor);
NbrCoor[d] = (Coor[d] + shifts[ii][d] + rd )%rd;
////////////////////////////////////////////////
// Inner index permute calculation
// Simpler version using icoor calculation
////////////////////////////////////////////////
SE._permute =0;
for(int d=0;d<Coor.size();d++){
int fd = grid->_fdimensions[d];
int rd = grid->_rdimensions[d];
int ly = grid->_simd_layout[d];
assert((ly==1)||(ly==2));
int shift = (shifts[ii][d]+fd)%fd; // make it strictly positive 0.. L-1
int x = Coor[d]; // x in [0... rd-1] as an oSite
int permute_dim = grid->PermuteDim(d);
int permute_slice=0;
if(permute_dim){
int num = shift%rd; // Slice within dest osite cell of slice zero
int wrap = shift/rd; // Number of osite local volume cells crossed through
// x+num < rd dictates whether we are in same permute state as slice 0
if ( x< rd-num ) permute_slice=wrap;
else permute_slice=(wrap+1)%ly;
}
if ( permute_slice ) {
int ptype =grid->PermuteType(d);
uint8_t mask =0x1<<ptype;
SE._permute |= mask;
}
}
////////////////////////////////////////////////
// Store in look up table
////////////////////////////////////////////////
this->_entries[lex] = SE;
} }
SE._offset = grid->oIndexReduced(NbrCoor); });
////////////////////////////////////////////////
// Inner index permute calculation
// Simpler version using icoor calculation
////////////////////////////////////////////////
SE._permute =0;
for(int d=0;d<Coor.size();d++){
int fd = grid->_fdimensions[d];
int rd = grid->_rdimensions[d];
int ly = grid->_simd_layout[d];
assert((ly==1)||(ly==2));
int shift = (shifts[ii][d]+fd)%fd; // make it strictly positive 0.. L-1
int x = Coor[d]; // x in [0... rd-1] as an oSite
int permute_dim = grid->PermuteDim(d);
int permute_slice=0;
if(permute_dim){
int num = shift%rd; // Slice within dest osite cell of slice zero
int wrap = shift/rd; // Number of osite local volume cells crossed through
// x+num < rd dictates whether we are in same permute state as slice 0
if ( x< rd-num ) permute_slice=wrap;
else permute_slice=(wrap+1)%ly;
}
if ( permute_slice ) {
int ptype =grid->PermuteType(d);
uint8_t mask =0x1<<ptype;
SE._permute |= mask;
}
}
////////////////////////////////////////////////
// Store in look up table
////////////////////////////////////////////////
this->_entries[lex] = SE;
}
}
} }
}; };