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

add template parameter to Smear_HISQ_fat for MILC interfacing

Browse Source
This commit is contained in:
david clarke 2023-06-20 15:37:54 -06:00
parent 8bb078db25
commit 26b2caf570
2 changed files with 20 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -71,6 +71,7 @@ struct SmearingParameters{
/*! @brief create fat links from link variables */ /*! @brief create fat links from link variables */
template<class LGF> // TODO: change to Gimpl?
class Smear_HISQ_fat { class Smear_HISQ_fat {
private: private:
@ -79,7 +80,8 @@ private:
public: public:
Smear_HISQ_fat(GridCartesian* grid, Real c1=1/8., Real cnaik=0., Real c3=1/16., Real c5=1/64., Real c7=1/384., Real clp=0.) // Don't allow default values here.
Smear_HISQ_fat(GridCartesian* grid, Real c1, Real cnaik, Real c3, Real c5, Real c7, Real clp)
: _grid(grid), : _grid(grid),
_LVL1(c1,cnaik,c3,c5,c7,clp) { _LVL1(c1,cnaik,c3,c5,c7,clp) {
assert(Nc == 3 && "HISQ smearing currently implemented only for Nc==3"); assert(Nc == 3 && "HISQ smearing currently implemented only for Nc==3");
@ -94,19 +96,21 @@ public:
~Smear_HISQ_fat() {} ~Smear_HISQ_fat() {}
void smear(LatticeGaugeField& u_smr, const LatticeGaugeField& U) const { void smear(LGF& u_smr, LGF& u_thin) const {
SmearingParameters lvl1 = this->_LVL1;
// Create a padded cell of extra padding depth=1 // Create a padded cell of extra padding depth=1
int depth = 1; int depth = 1;
PaddedCell Ghost(depth,this->_grid); PaddedCell Ghost(depth,this->_grid);
LatticeGaugeField Ughost = Ghost.Exchange(u_smr); LGF Ughost = Ghost.Exchange(u_thin);
// Array for <tr U_mu_nu>(x) // Array for <tr U_mu_nu>(x)
GridBase *GhostGrid = Ughost.Grid(); GridBase *GhostGrid = Ughost.Grid();
LatticeComplex gplaq(GhostGrid); LatticeComplex gplaq(GhostGrid);
// This is where the 3-link constructs will be stored // This is where the 3-link constructs will be stored
LatticeGaugeField Ughost_fat(Ughost.Grid()); LGF Ughost_fat(Ughost.Grid());
// Create 3-link stencil (class will build its own stencils) // Create 3-link stencil (class will build its own stencils)
// writing your own stencil, you're hard-coding the periodic BCs, so you don't need // writing your own stencil, you're hard-coding the periodic BCs, so you don't need
@ -136,7 +140,7 @@ public:
Ughost_fat=Zero(); Ughost_fat=Zero();
// Create the accessors, here U_v and U_fat_v // Create the accessors, here U_v and U_fat_v
autoView(U_v , Ughost , CpuRead); autoView(U_v , Ughost , CpuRead);
autoView(U_fat_v, Ughost_fat, CpuWrite); autoView(U_fat_v, Ughost_fat, CpuWrite);
// This is a loop over local sites. // This is a loop over local sites.
@ -186,12 +190,8 @@ public:
gpermute(U4,SE4->_permute); gpermute(U4,SE4->_permute);
gpermute(U5,SE5->_permute); gpermute(U5,SE5->_permute);
// Forward contribution from this orientation // forward backward
auto W = U0*U1*U2; auto W = U0*U1*U2 + U3*U4*U5;
U_fat_v[ss](mu) = U_fat_v[ss](mu) + W;
// Backward contribution from this orientation
W = U3*U4*U5;
U_fat_v[ss](mu) = U_fat_v[ss](mu) + W; U_fat_v[ss](mu) = U_fat_v[ss](mu) + W;
s=s+6; s=s+6;
@ -199,14 +199,9 @@ public:
} }
} }
// Here is my understanding of this part: The padded cell has its own periodic BCs, so u_smr = lvl1.c_3*Ghost.Extract(Ughost_fat) + lvl1.c_1*u_thin;
// if I take a step to the right at the right-most side of the cell, I end up on the
// left-most side. This means that the plaquettes in the padding are wrong. Luckily
// all we care about are the plaquettes in the cell, which we obtain from Extract.
u_smr = Ghost.Extract(Ughost_fat);
}; };
// I guess the way this will go is: // I guess the way this will go is:
// 1. 3-link smear // 1. 3-link smear
// 2. exchange // 2. exchange
@ -219,6 +214,7 @@ public:
/*! @brief create long links from link variables. */ /*! @brief create long links from link variables. */
template<class LGF>
class Smear_HISQ_Naik { class Smear_HISQ_Naik {
private: private:
@ -233,16 +229,16 @@ public:
~Smear_HISQ_Naik() {} ~Smear_HISQ_Naik() {}
void smear(LatticeGaugeField& u_smr, const LatticeGaugeField& U) const { void smear(LGF& u_smr, const LGF& U) const {
int depth = 1; int depth = 1;
PaddedCell Ghost(depth,this->_grid); PaddedCell Ghost(depth,this->_grid);
LatticeGaugeField Ughost = Ghost.Exchange(u_smr); LGF Ughost = Ghost.Exchange(u_smr);
GridBase *GhostGrid = Ughost.Grid(); GridBase *GhostGrid = Ughost.Grid();
LatticeComplex gplaq(GhostGrid); LatticeComplex gplaq(GhostGrid);
LatticeGaugeField Ughost_naik(Ughost.Grid()); LGF Ughost_naik(Ughost.Grid());
std::vector<Coordinate> shifts; std::vector<Coordinate> shifts;
for(int mu=0;mu<Nd;mu++){ for(int mu=0;mu<Nd;mu++){

7
tests/smearing/Test_fatLinks.cc
View File

@ -70,15 +70,14 @@ int main (int argc, char **argv)
NerscIO::readConfiguration(Umu, header, param.conf_in); NerscIO::readConfiguration(Umu, header, param.conf_in);
// Smear Umu and store result in U_smr // Smear Umu and store result in U_smr
Smear_HISQ_fat hisq_fat(&GRID); Smear_HISQ_fat<LatticeGaugeField> hisq_fat(&GRID,1/8.,0.,1/16.,1/64.,1/384.,0.);
hisq_fat.smear(U_smr,Umu); hisq_fat.smear(U_smr,Umu);
NerscIO::writeConfiguration(U_smr,param.conf_out,"HISQ"); NerscIO::writeConfiguration(U_smr,param.conf_out,"HISQ");
// In the following, we test instantiation of Smear_HISQ_fat in different ways: // Test a C-style instantiation
Smear_HISQ_fat hisq_fat_typical(&GRID,1,2,3,4,5,6);
double path_coeff[6] = {1, 2, 3, 4, 5, 6}; double path_coeff[6] = {1, 2, 3, 4, 5, 6};
Smear_HISQ_fat hisq_fat_Cstyle(&GRID,path_coeff); Smear_HISQ_fat<LatticeGaugeField> hisq_fat_Cstyle(&GRID,path_coeff);
Grid_finalize(); Grid_finalize();
} }