portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-03 18:55:56 +01:00
Code Releases Activity

now compiles

Browse Source
This commit is contained in:
david clarke 2023-06-09 16:27:45 -06:00
parent 4b994a1bc7
commit 1cf9ec1cce
2 changed files with 149 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -51,25 +51,21 @@ template<class vobj> void gpermute(vobj & inout,int perm) {
}
/*! @brief 3-link smearing of link variable. */
//template <class Gimpl>
//class Smear_HISQ_3link : public Smear<Gimpl> {
class Smear_HISQ_3link {
// TODO: I'm not using Gimpl so I don't know how to inherit
/*! @brief create fat links from link variables. */
class Smear_HISQ_fat {
private:
// std::vector<int> _linkTreatment;
GridBase* const _grid;
GridCartesian* const _grid;
public:
// INHERIT_GIMPL_TYPES(Gimpl)
// Eventually this will take, e.g., coefficients as argument
Smear_HISQ_3link(GridBase* grid) : _grid(grid) {
Smear_HISQ_fat(GridCartesian* grid) : _grid(grid) {
assert(Nc == 3 && "HISQ smearing currently implemented only for Nc==3");
}
~Smear_HISQ_3link() {}
~Smear_HISQ_fat() {}
void smear(LatticeGaugeField& u_smr, const LatticeGaugeField& U) const {
@ -83,7 +79,7 @@ public:
LatticeComplex gplaq(GhostGrid);
// This is where the 3-link constructs will be stored
LatticeGaugeField Ughost_3link(Ughost.Grid());
LatticeGaugeField Ughost_fat(Ughost.Grid());
// 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
@ -110,11 +106,11 @@ public:
}
GeneralLocalStencil gStencil(GhostGrid,shifts);
Ughost_3link=Zero();
Ughost_fat=Zero();
// Create the accessors, here U_v and U_3link_v
// Create the accessors, here U_v and U_fat_v
autoView(U_v , Ughost , CpuRead);
autoView(U_3link_v, Ughost_3link, CpuWrite);
autoView(U_fat_v, Ughost_fat, CpuWrite);
// This is a loop over local sites.
for(int ss=0;ss<U_v.size();ss++){
@ -165,11 +161,11 @@ public:
// Forward contribution from this orientation
auto W = U0*U1*U2;
U_3link_v[ss](mu) = U_3link_v[ss](mu) + W;
U_fat_v[ss](mu) = U_fat_v[ss](mu) + W;
// Backward contribution from this orientation
W = U3*U4*U5;
U_3link_v[ss](mu) = U_3link_v[ss](mu) + W;
U_fat_v[ss](mu) = U_fat_v[ss](mu) + W;
s=s+6;
}
@ -180,11 +176,145 @@ public:
// 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_3link);
u_smr = Ghost.Extract(Ughost_fat);
};
// I guess the way this will go is:
// 1. 3-link smear
// 2. exchange
// 3. 5-link calculated from 3-link
// void derivative(const GaugeField& Gauge) const {
// };
};
/*! @brief create long links from link variables. */
class Smear_HISQ_Naik {
private:
GridCartesian* const _grid;
public:
// Eventually this will take, e.g., coefficients as argument
Smear_HISQ_Naik(GridCartesian* grid) : _grid(grid) {
assert(Nc == 3 && "HISQ smearing currently implemented only for Nc==3");
}
~Smear_HISQ_Naik() {}
void smear(LatticeGaugeField& u_smr, const LatticeGaugeField& U) const {
int depth = 1;
PaddedCell Ghost(depth,this->_grid);
LatticeGaugeField Ughost = Ghost.Exchange(u_smr);
GridBase *GhostGrid = Ughost.Grid();
LatticeComplex gplaq(GhostGrid);
LatticeGaugeField Ughost_naik(Ughost.Grid());
std::vector<Coordinate> shifts;
for(int mu=0;mu<Nd;mu++){
for(int nu=mu+1;nu<Nd;nu++){
// forward shifts
Coordinate shift_0(Nd,0);
Coordinate shift_mu(Nd,0); shift_mu[mu]=1;
Coordinate shift_nu(Nd,0); shift_nu[nu]=1;
// push_back creates an element at the end of shifts and
// assigns the data in the argument to it.
shifts.push_back(shift_mu);
shifts.push_back(shift_nu);
shifts.push_back(shift_0);
// reverse shifts
shift_nu[nu]=-1;
Coordinate shift_munu(Nd,0); shift_munu[mu]=1; shift_munu[nu]=-1;
shifts.push_back(shift_munu);
shifts.push_back(shift_nu); // in principle you don't need both of these grid points,
shifts.push_back(shift_nu); // but it helps the reader keep track of offsets
}
}
GeneralLocalStencil gStencil(GhostGrid,shifts);
Ughost_naik=Zero();
// Create the accessors, here U_v and U_fat_v
autoView(U_v , Ughost , CpuRead);
autoView(U_naik_v, Ughost_naik, CpuWrite);
// This is a loop over local sites.
for(int ss=0;ss<U_v.size();ss++){
// This is the stencil index. It increases as we make our way through the spacetime sites,
// plaquette orientations, and as we travel around a plaquette.
int s=0;
for(int mu=0;mu<Nd;mu++){
for(int nu=mu+1;nu<Nd;nu++){
// shift_mu; shift_mu[mu]=1
// shift_nu; shift_nu[nu]=1
// shift_0
// shift_munu; shift_munu[mu]= 1; shift_munu[nu]=-1;
// shift_nu ; shift_nu[nu]=-1;
// shift_nu ; shift_nu[nu]=-1;
auto SE0 = gStencil.GetEntry(s+0,ss);
auto SE1 = gStencil.GetEntry(s+1,ss);
auto SE2 = gStencil.GetEntry(s+2,ss);
auto SE3 = gStencil.GetEntry(s+3,ss);
auto SE4 = gStencil.GetEntry(s+4,ss);
auto SE5 = gStencil.GetEntry(s+5,ss);
// Each offset corresponds to a site around the plaquette.
int o0 = SE0->_offset;
int o1 = SE1->_offset;
int o2 = SE2->_offset;
int o3 = SE3->_offset;
int o4 = SE4->_offset;
int o5 = SE5->_offset;
auto U0 = U_v[o0](nu);
auto U1 = adj(U_v[o1](mu));
auto U2 = adj(U_v[o2](nu));
gpermute(U0,SE0->_permute);
gpermute(U1,SE1->_permute);
gpermute(U2,SE2->_permute);
auto U3 = adj(U_v[o3](nu));
auto U4 = adj(U_v[o4](mu));
auto U5 = U_v[o5](nu);
gpermute(U3,SE3->_permute);
gpermute(U4,SE4->_permute);
gpermute(U5,SE5->_permute);
// Forward contribution from this orientation
auto W = U0*U1*U2;
U_naik_v[ss](mu) = U_naik_v[ss](mu) + W;
// Backward contribution from this orientation
W = U3*U4*U5;
U_naik_v[ss](mu) = U_naik_v[ss](mu) + W;
s=s+6;
}
}
}
// Here is my understanding of this part: The padded cell has its own periodic BCs, so
// 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_naik);
};
// void derivative(const GaugeField& Gauge) const {
// };
};
NAMESPACE_END(Grid);

4
tests/smearing/Test_fatLinks.cc
View File

@ -67,9 +67,9 @@ int main (int argc, char **argv)
FieldMetaData header;
NerscIO::readConfiguration(Umu, header, param.conf_in);
Smear_HISQ_3link hisq_3link(&GRID);
Smear_HISQ_fat hisq_fat(&GRID);
hisq_3link.smear(U_smr,Umu);
hisq_fat.smear(U_smr,Umu);
NerscIO::writeConfiguration(U_smr,param.conf_out,"HISQ");