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

Coordinate handling GPU friendly

Browse Source
This commit is contained in:
paboyle 2018-02-24 22:21:36 +00:00
parent 7574c18cef
commit bf5fb89aff
3 changed files with 19 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
lib/stencil/Lebesgue.cc
View File

@ -100,9 +100,9 @@ void LebesgueOrder::CartesianBlocking(void)
assert(ND==4);
assert(ND==Block.size());
std::vector<IndexInteger> dims(ND);
std::vector<IndexInteger> xo(ND,0);
std::vector<IndexInteger> xi(ND,0);
Coordinate dims(ND);
Coordinate xo(ND,0);
Coordinate xi(ND,0);
for(IndexInteger mu=0;mu<ND;mu++){
dims[mu] = grid->_rdimensions[mu];
@ -112,9 +112,9 @@ void LebesgueOrder::CartesianBlocking(void)
};
void LebesgueOrder::IterateO(int ND,int dim,
std::vector<IndexInteger> & xo,
std::vector<IndexInteger> & xi,
std::vector<IndexInteger> &dims)
Coordinate & xo,
Coordinate & xi,
Coordinate &dims)
{
for(xo[dim]=0;xo[dim]<dims[dim];xo[dim]+=Block[dim]){
if ( dim > 0 ) {
@ -127,11 +127,11 @@ void LebesgueOrder::IterateO(int ND,int dim,
void LebesgueOrder::IterateI(int ND,
int dim,
std::vector<IndexInteger> & xo,
std::vector<IndexInteger> & xi,
std::vector<IndexInteger> &dims)
Coordinate & xo,
Coordinate & xi,
Coordinate &dims)
{
std::vector<IndexInteger> x(ND);
Coordinate x(ND);
for(xi[dim]=0;xi[dim]<std::min(dims[dim]-xo[dim],Block[dim]);xi[dim]++){
if ( dim > 0 ) {
IterateI(ND,dim-1,xo,xi,dims);
@ -157,8 +157,8 @@ void LebesgueOrder::ZGraph(void)
const IndexInteger one=1;
IndexInteger ND = grid->_ndimension;
std::vector<IndexInteger> dims(ND);
std::vector<IndexInteger> adims(ND);
Coordinate dims(ND);
Coordinate adims(ND);
std::vector<std::vector<IndexInteger> > bitlist(ND);
for(IndexInteger mu=0;mu<ND;mu++){

13
lib/stencil/Lebesgue.h
View File

@ -38,7 +38,6 @@ class LebesgueOrder {
public:
typedef int32_t IndexInteger;
static int UseLebesgueOrder;
GridBase *grid;
@ -62,13 +61,13 @@ public:
void NoBlocking(void);
void CartesianBlocking(void);
void IterateO(int ND,int dim,
std::vector<IndexInteger> & xo,
std::vector<IndexInteger> & xi,
std::vector<IndexInteger> &dims);
Coordinate & xo,
Coordinate & xi,
Coordinate &dims);
void IterateI(int ND,int dim,
std::vector<IndexInteger> & xo,
std::vector<IndexInteger> & xi,
std::vector<IndexInteger> &dims);
Coordinate & xo,
Coordinate & xi,
Coordinate &dims);
void ThreadInterleave(void);

2
lib/stencil/Stencil.h
View File

@ -150,7 +150,7 @@ public:
std::vector<int> _distances;
std::vector<int> _comm_buf_size;
std::vector<int> _permute_type;
std::vector<int> _simd_layout;
Coordinate _simd_layout;
Vector<StencilEntry> _entries; // Resident in managed memory
StencilEntry* _entries_p;