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

try 7-link

Browse Source
This commit is contained in:
david clarke 2023-09-16 23:18:16 -06:00
parent affff3865f
commit d247031c98
2 changed files with 67 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
Grid/qcd/smearing/HISQSmearing.h
View File

@ -129,7 +129,7 @@ struct SmearingParameters{
/*! @brief create fat links from link variables */ /*! @brief create fat links from link variables */
template<class LGF> // TODO: change to Gimpl? template<class LGF>
class Smear_HISQ_fat { class Smear_HISQ_fat {
private: private:
@ -168,6 +168,8 @@ public:
// This is where auxiliary N-link fields and the final smear will be stored. // This is where auxiliary N-link fields and the final smear will be stored.
LGF Ughost_fat(Ughost.Grid()); LGF Ughost_fat(Ughost.Grid());
LGF Ughost_3link(Ughost.Grid()); LGF Ughost_3link(Ughost.Grid());
LGF Ughost_5linkA(Ughost.Grid());
LGF Ughost_5linkB(Ughost.Grid());
// Create 3-link stencil. We allow mu==nu just to make the indexing easier. // Create 3-link stencil. We allow mu==nu just to make the indexing easier.
// Shifts with mu==nu will not be used. // Shifts with mu==nu will not be used.
@ -188,13 +190,18 @@ public:
Ughost_fat=Zero(); Ughost_fat=Zero();
// Create the accessors // Create the accessors
autoView(U_v , Ughost , CpuRead); autoView(U_v , Ughost , CpuRead);
autoView(U_fat_v , Ughost_fat , CpuWrite); autoView(U_fat_v , Ughost_fat , CpuWrite);
autoView(U_3link_v, Ughost_3link, CpuWrite); autoView(U_3link_v , Ughost_3link , CpuWrite);
autoView(U_5linkA_v, Ughost_5linkA, CpuWrite);
autoView(U_5linkB_v, Ughost_5linkB, CpuWrite);
for(int mu=0;mu<Nd;mu++) { for(int mu=0;mu<Nd;mu++) {
Ughost_3link=Zero(); Ughost_3link =Zero();
Ughost_5linkA=Zero();
Ughost_5linkB=Zero();
// 3-link // 3-link
for(int site=0;site<U_v.size();site++){ for(int site=0;site<U_v.size();site++){
@ -230,12 +237,15 @@ public:
} }
} }
// 5-link // 5-link
for(int site=0;site<U_v.size();site++){ for(int site=0;site<U_v.size();site++){
int sigmaIndex = 0;
for(int nu=0;nu<Nd;nu++) { for(int nu=0;nu<Nd;nu++) {
if(nu==mu) continue; if(nu==mu) continue;
int s = stencilIndex(mu,nu); int s = stencilIndex(mu,nu);
for(int rho=0;rho<Nd;rho++) { for(int rho=0;rho<Nd;rho++) {
if(rho==mu) continue;
if(rho==nu) continue; if(rho==nu) continue;
auto SE0 = gStencil.GetEntry(s+0,site); int x_p_mu = SE0->_offset; auto SE0 = gStencil.GetEntry(s+0,site); int x_p_mu = SE0->_offset;
@ -253,7 +263,57 @@ public:
auto W = U2*U1*adj(U0) + adj(U5)*U4*U3; auto W = U2*U1*adj(U0) + adj(U5)*U4*U3;
if(sigmaIndex<3) {
U_5linkA_v[site](rho) = W;
} else {
U_5linkB_v[site](rho) = W;
}
U_fat_v[site](mu) = U_fat_v[site](mu) + lt.c_5*W; U_fat_v[site](mu) = U_fat_v[site](mu) + lt.c_5*W;
sigmaIndex++;
}
}
}
// 7-link
for(int site=0;site<U_v.size();site++){
int sigmaIndex = 0;
for(int nu=0;nu<Nd;nu++) {
if(nu==mu) continue;
int s = stencilIndex(mu,nu);
for(int rho=0;rho<Nd;rho++) {
if(rho==mu) continue;
if(rho==nu) continue;
auto SE0 = gStencil.GetEntry(s+0,site); int x_p_mu = SE0->_offset;
auto SE1 = gStencil.GetEntry(s+1,site); int x_p_nu = SE1->_offset;
auto SE2 = gStencil.GetEntry(s+2,site); int x = SE2->_offset;
auto SE3 = gStencil.GetEntry(s+3,site); int x_p_mu_m_nu = SE3->_offset;
auto SE4 = gStencil.GetEntry(s+4,site); int x_m_nu = SE4->_offset;
auto U0 = U_v[x_p_mu ](nu) ; gpermute(U0,SE0->_permute);
auto U1 = U0;
if(sigmaIndex<3) {
U1 = U_5linkB_v[x_p_nu](rho); gpermute(U1,SE1->_permute);
} else {
U1 = U_5linkA_v[x_p_nu](rho); gpermute(U1,SE1->_permute);
}
auto U2 = U_v[x ](nu) ; gpermute(U2,SE2->_permute);
auto U3 = U_v[x_p_mu_m_nu](nu) ; gpermute(U3,SE3->_permute);
auto U4 = U0;
if(sigmaIndex<3) {
U4 = U_5linkB_v[x_m_nu](rho); gpermute(U4,SE4->_permute);
} else {
U4 = U_5linkA_v[x_m_nu](rho); gpermute(U4,SE4->_permute);
}
auto U5 = U_v[x_m_nu ](nu) ; gpermute(U5,SE4->_permute);
auto W = U2*U1*adj(U0) + adj(U5)*U4*U3;
U_fat_v[site](mu) = U_fat_v[site](mu) + lt.c_7*W;
sigmaIndex++;
} }
} }
} }

5
tests/smearing/Test_fatLinks.cc
View File

@ -78,7 +78,7 @@ int main (int argc, char** argv) {
int Nt = 4; int Nt = 4;
Coordinate latt_size(Nd,0); latt_size[0]=Ns; latt_size[1]=Ns; latt_size[2]=Ns; latt_size[3]=Nt; Coordinate latt_size(Nd,0); latt_size[0]=Ns; latt_size[1]=Ns; latt_size[2]=Ns; latt_size[3]=Nt;
std::string conf_in = "nersc.l8t4b3360"; std::string conf_in = "nersc.l8t4b3360";
std::string conf_out = "nersc.l8t4b3360.3link"; std::string conf_out = "nersc.l8t4b3360.35link";
int threads = GridThread::GetThreads(); int threads = GridThread::GetThreads();
// Initialize the Grid // Initialize the Grid
@ -105,7 +105,6 @@ int main (int argc, char** argv) {
// Smear Umu and store result in U_smr // Smear Umu and store result in U_smr
Smear_HISQ_fat<LatticeGaugeField> hisq_fat(&GRID,1/8.,0.,1/16.,1/64.,1/384.,0.); Smear_HISQ_fat<LatticeGaugeField> hisq_fat(&GRID,1/8.,0.,1/16.,1/64.,1/384.,0.);
// Smear_HISQ_fat<LatticeGaugeField> hisq_fat(&GRID,1/8.,0.,1/16.,0.,1/384.,0.);
hisq_fat.smear(U_smr,Umu); hisq_fat.smear(U_smr,Umu);
NerscIO::writeConfiguration(U_smr,conf_out,"HISQ"); NerscIO::writeConfiguration(U_smr,conf_out,"HISQ");
@ -115,7 +114,7 @@ int main (int argc, char** argv) {
Smear_HISQ_fat<LatticeGaugeField> hisq_fat_Cstyle(&GRID,path_coeff); Smear_HISQ_fat<LatticeGaugeField> hisq_fat_Cstyle(&GRID,path_coeff);
// Make sure result doesn't change w.r.t. a trusted lattice // Make sure result doesn't change w.r.t. a trusted lattice
NerscIO::readConfiguration(Umu, header, "nersc.l8t4b3360.3link.control"); NerscIO::readConfiguration(Umu, header, "nersc.l8t4b3360.35link.control");
LatticeGaugeField diff(&GRID); LatticeGaugeField diff(&GRID);
diff = Umu-U_smr; diff = Umu-U_smr;
auto absDiff = norm2(diff)/norm2(Umu); auto absDiff = norm2(diff)/norm2(Umu);