portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2024-11-09 23:45:36 +00:00
Code Releases Activity

7-link working

Browse Source
This commit is contained in:
david clarke 2023-10-10 22:41:52 -06:00
parent 63d9b8e8a3
commit 0cfd13d18b
2 changed files with 18 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -82,7 +82,7 @@ void generalShift(Coordinate& shift, int dir) {
} }
} }
// Move into general stencil header, beneath definition of general stencil
/*! @brief follow a path of directions, shifting one unit in each direction */ /*! @brief follow a path of directions, shifting one unit in each direction */
template<typename... Args> template<typename... Args>
void generalShift(Coordinate& shift, int dir, Args... args) { void generalShift(Coordinate& shift, int dir, Args... args) {
@ -118,8 +118,8 @@ struct SmearingParameters{
Real c_5; // 5 link Real c_5; // 5 link
Real c_7; // 7 link Real c_7; // 7 link
Real c_lp; // 5 link Lepage Real c_lp; // 5 link Lepage
SmearingParameters(Real c1, Real cnaik, Real c3, Real c5, Real c7, Real clp) : SmearingParameters(Real c1, Real cnaik, Real c3, Real c5, Real c7, Real clp)
c_1(c1), : c_1(c1),
c_naik(cnaik), c_naik(cnaik),
c_3(c3), c_3(c3),
c_5(c5), c_5(c5),
@ -198,11 +198,12 @@ public:
for(int mu=0;mu<Nd;mu++) { for(int mu=0;mu<Nd;mu++) {
// TODO: This approach is slightly memory inefficient. It uses 25% more memory than
// needs to be used.
Ughost_3link =Zero(); Ughost_3link =Zero();
Ughost_5linkA=Zero(); Ughost_5linkA=Zero();
Ughost_5linkB=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++){
for(int nu=0;nu<Nd;nu++) { for(int nu=0;nu<Nd;nu++) {
@ -245,8 +246,7 @@ public:
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 == mu || 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;
auto SE1 = gStencil.GetEntry(s+1,site); int x_p_nu = SE1->_offset; auto SE1 = gStencil.GetEntry(s+1,site); int x_p_nu = SE1->_offset;
@ -254,6 +254,8 @@ public:
auto SE3 = gStencil.GetEntry(s+3,site); int x_p_mu_m_nu = SE3->_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 SE4 = gStencil.GetEntry(s+4,site); int x_m_nu = SE4->_offset;
// gpermutes will be replaced with single line of code, combines load and permute
// into one step. still in pull request stage
auto U0 = U_v[x_p_mu ](nu) ; gpermute(U0,SE0->_permute); auto U0 = U_v[x_p_mu ](nu) ; gpermute(U0,SE0->_permute);
auto U1 = U_3link_v[x_p_nu ](rho); gpermute(U1,SE1->_permute); auto U1 = U_3link_v[x_p_nu ](rho); gpermute(U1,SE1->_permute);
auto U2 = U_v[x ](nu) ; gpermute(U2,SE2->_permute); auto U2 = U_v[x ](nu) ; gpermute(U2,SE2->_permute);
@ -283,8 +285,7 @@ public:
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 == mu || 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;
auto SE1 = gStencil.GetEntry(s+1,site); int x_p_nu = SE1->_offset; auto SE1 = gStencil.GetEntry(s+1,site); int x_p_nu = SE1->_offset;
@ -293,6 +294,7 @@ public:
auto SE4 = gStencil.GetEntry(s+4,site); int x_m_nu = SE4->_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 U0 = U_v[x_p_mu ](nu) ; gpermute(U0,SE0->_permute);
// decltype, or auto U1 = { ? ... }
auto U1 = U0; auto U1 = U0;
if(sigmaIndex<3) { if(sigmaIndex<3) {
U1 = U_5linkB_v[x_p_nu](rho); gpermute(U1,SE1->_permute); U1 = U_5linkB_v[x_p_nu](rho); gpermute(U1,SE1->_permute);
@ -311,6 +313,7 @@ public:
auto W = U2*U1*adj(U0) + adj(U5)*U4*U3; auto W = U2*U1*adj(U0) + adj(U5)*U4*U3;
// std::vector<LatticeColorMatrix>(3) ?
U_fat_v[site](mu) = U_fat_v[site](mu) + lt.c_7*W; U_fat_v[site](mu) = U_fat_v[site](mu) + lt.c_7*W;
sigmaIndex++; sigmaIndex++;

4
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.35link"; std::string conf_out = "nersc.l8t4b3360.357link";
int threads = GridThread::GetThreads(); int threads = GridThread::GetThreads();
// Initialize the Grid // Initialize the Grid
@ -114,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.35link.control"); NerscIO::readConfiguration(Umu, header, "nersc.l8t4b3360.357link.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);