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

View related changes

Browse Source
This commit is contained in:
paboyle 2018-03-04 16:34:14 +00:00
parent 8824a54269
commit 442b0b406c
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
lib/qcd/action/scalar/ScalarInteractionAction.h
View File

@ -78,10 +78,12 @@ public:
static Stencil phiStencil(p.Grid(), npoint, 0, directions, displacements); static Stencil phiStencil(p.Grid(), npoint, 0, directions, displacements);
phiStencil.HaloExchange(p, compressor); phiStencil.HaloExchange(p, compressor);
Field action(p.Grid()), pshift(p.Grid()), phisquared(p.Grid()); Field action(p.Grid()), pshift(p.Grid()), phisquared(p.Grid());
auto action_v = action.View();
phisquared = p*p; phisquared = p*p;
action = (2.0*Ndim + mass_square)*phisquared - lambda/24.*phisquared*phisquared; action = (2.0*Ndim + mass_square)*phisquared - lambda/24.*phisquared*phisquared;
for (int mu = 0; mu < Ndim; mu++) { for (int mu = 0; mu < Ndim; mu++) {
// pshift = Cshift(p, mu, +1); // not efficient, implement with stencils // pshift = Cshift(p, mu, +1); // not efficient, implement with stencils
auto p_v = p.View();
thread_loop( (int i = 0; i < p.Grid()->oSites(); i++) ,{ thread_loop( (int i = 0; i < p.Grid()->oSites(); i++) ,{
int permute_type; int permute_type;
StencilEntry *SE; StencilEntry *SE;
@ -89,17 +91,17 @@ public:
const vobj *temp, *t_p; const vobj *temp, *t_p;
SE = phiStencil.GetEntry(permute_type, mu, i); SE = phiStencil.GetEntry(permute_type, mu, i);
t_p = &p[i]; t_p = &p_v[i];
if ( SE->_is_local ) { if ( SE->_is_local ) {
temp = &p[SE->_offset]; temp = &p_v[SE->_offset];
if ( SE->_permute ) { if ( SE->_permute ) {
permute(temp2, *temp, permute_type); permute(temp2, *temp, permute_type);
action[i] -= temp2*(*t_p) + (*t_p)*temp2; action_v[i] -= temp2*(*t_p) + (*t_p)*temp2;
} else { } else {
action[i] -= (*temp)*(*t_p) + (*t_p)*(*temp); action_v[i] -= (*temp)*(*t_p) + (*t_p)*(*temp);
} }
} else { } else {
action[i] -= phiStencil.CommBuf()[SE->_offset]*(*t_p) + (*t_p)*phiStencil.CommBuf()[SE->_offset]; action_v[i] -= phiStencil.CommBuf()[SE->_offset]*(*t_p) + (*t_p)*phiStencil.CommBuf()[SE->_offset];
} }
}); });
// action -= pshift*p + p*pshift; // action -= pshift*p + p*pshift;
@ -118,6 +120,8 @@ public:
//for (int mu = 0; mu < Nd; mu++) force -= Cshift(p, mu, -1) + Cshift(p, mu, 1); //for (int mu = 0; mu < Nd; mu++) force -= Cshift(p, mu, -1) + Cshift(p, mu, 1);
for (int point = 0; point < npoint; point++) { for (int point = 0; point < npoint; point++) {
auto force_v = force.View();
auto p_v = p.View();
thread_loop( (int i = 0; i < p.Grid()->oSites(); i++) ,{ thread_loop( (int i = 0; i < p.Grid()->oSites(); i++) ,{
const vobj *temp; const vobj *temp;
vobj temp2; vobj temp2;
@ -126,15 +130,15 @@ public:
SE = phiStencil.GetEntry(permute_type, point, i); SE = phiStencil.GetEntry(permute_type, point, i);
if ( SE->_is_local ) { if ( SE->_is_local ) {
temp = &p[SE->_offset]; temp = &p_v[SE->_offset];
if ( SE->_permute ) { if ( SE->_permute ) {
permute(temp2, *temp, permute_type); permute(temp2, *temp, permute_type);
force[i] -= temp2; force_v[i] -= temp2;
} else { } else {
force[i] -= *temp; force_v[i] -= *temp;
} }
} else { } else {
force[i] -= phiStencil.CommBuf()[SE->_offset]; force_v[i] -= phiStencil.CommBuf()[SE->_offset];
} }
}); });
} }