portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-09 21:50:45 +01:00
Code Releases Activity

Simplified tests

Browse Source
This commit is contained in:
Michael Marshall 2019-01-24 18:50:18 +00:00
parent f92ed659a7
commit 577cdf1d72
2 changed files with 78 additions and 296 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

107
Hadrons/Modules/MDistil/LapEvec.hpp
View File

@ -79,17 +79,17 @@ struct LanczosParameters: Serializable {
template <class ReaderClass> LanczosParameters(Reader<ReaderClass>& Reader){read(Reader,"Lanczos",*this);} template <class ReaderClass> LanczosParameters(Reader<ReaderClass>& Reader){read(Reader,"Lanczos",*this);}
}; };
struct DistilParameters: Serializable { /*struct DistilParameters: Serializable {
GRID_SERIALIZABLE_CLASS_MEMBERS(DistilParameters, GRID_SERIALIZABLE_CLASS_MEMBERS(DistilParameters
int, TI, ,int, TI
int, LI, ,int, LI
int, Nnoise, ,int, Nnoise
int, Ls, // For makeFiveDimGrid ,int, tSrc
int, tSrc) )//,int, Ls) // For makeFiveDimGrid
DistilParameters() = default; DistilParameters() = default;
template <class ReaderClass> DistilParameters(Reader<ReaderClass>& Reader){read(Reader,"Distil",*this);} template <class ReaderClass> DistilParameters(Reader<ReaderClass>& Reader){read(Reader,"Distil",*this);}
}; };
struct SolverParameters: Serializable { struct SolverParameters: Serializable {
GRID_SERIALIZABLE_CLASS_MEMBERS(SolverParameters, GRID_SERIALIZABLE_CLASS_MEMBERS(SolverParameters,
double, CGPrecision, double, CGPrecision,
@ -98,21 +98,21 @@ struct SolverParameters: Serializable {
double, M5) double, M5)
SolverParameters() = default; SolverParameters() = default;
template <class ReaderClass> SolverParameters(Reader<ReaderClass>& Reader){read(Reader,"Solver",*this);} template <class ReaderClass> SolverParameters(Reader<ReaderClass>& Reader){read(Reader,"Solver",*this);}
}; };*/
// These are the actual parameters passed to the module during construction // These are the actual parameters passed to the module during construction
class LapEvecPar: Serializable class LapEvecPar: Serializable
{ {
public: public:
GRID_SERIALIZABLE_CLASS_MEMBERS(LapEvecPar, GRID_SERIALIZABLE_CLASS_MEMBERS(LapEvecPar
std::string, gauge, ,std::string, gauge
std::string, EigenPackName, //,std::string, EigenPackName
StoutParameters, Stout, ,StoutParameters, Stout
ChebyshevParameters, Cheby, ,ChebyshevParameters, Cheby
LanczosParameters, Lanczos, ,LanczosParameters, Lanczos
DistilParameters, Distil, //,DistilParameters, Distil
SolverParameters, Solver); )//,SolverParameters, Solver)
}; };
/****************************************************************************** /******************************************************************************
@ -231,69 +231,69 @@ void TLapEvec<FImpl>::Cleanup(void)
template <typename FImpl> template <typename FImpl>
void TLapEvec<FImpl>::execute(void) void TLapEvec<FImpl>::execute(void)
{ {
LOG(Message) << "execute() : start" << std::endl; LOG(Message) << "execute() : start for " << getName() << std::endl;
// Alii for parameters // Alii for parameters
const int &TI{par().Distil.TI}; //const int &TI{par().Distil.TI};
const int &LI{par().Distil.LI}; //const int &LI{par().Distil.LI};
const int &nnoise{par().Distil.Nnoise}; //const int &nnoise{par().Distil.Nnoise};
const int &tsrc{par().Distil.tSrc}; //const int &tsrc{par().Distil.tSrc};
const LanczosParameters &LPar{par().Lanczos};
const int &nvec{LPar.Nvec};
const bool exact_distillation{TI==Nt && LI==nvec};
const bool full_tdil{TI==Nt};
const int &Nt_inv{full_tdil ? 1 : TI};
const ChebyshevParameters &ChebPar{par().Cheby}; const ChebyshevParameters &ChebPar{par().Cheby};
const LanczosParameters &LPar{par().Lanczos};
const int &nvec{LPar.Nvec};
//const bool exact_distillation{TI==Nt && LI==nvec};
//const bool full_tdil{TI==Nt};
//const int &Nt_inv{full_tdil ? 1 : TI};
// Assertions on the parameters we read // Assertions on the parameters we read
assert(TI>1); //assert(TI>1);
assert(LI>1); //assert(LI>1);
if(exact_distillation) //if(exact_distillation)
assert(nnoise==1); //assert(nnoise==1);
else //else
assert(nnoise>1); //assert(nnoise>1);
// Debugging only // Debugging only
//envGetTmp(GaugeField, Umu); //envGetTmp(GaugeField, Umu);
auto &Umu = envGet(GaugeField, par().gauge); auto &Umu = envGet(GaugeField, par().gauge);
if((1)) { envGetTmp(GaugeField, Umu_smear);
const std::vector<int> seeds({1, 2, 3, 4, 5}); if((0)) {
GridParallelRNG pRNG4d(gridHD); //const std::vector<int> seeds({1, 2, 3, 4, 5});
pRNG4d.SeedFixedIntegers(seeds); //GridParallelRNG pRNG4d(gridHD);
std::cout << GridLogMessage << "now hot config" << std::endl; //pRNG4d.SeedFixedIntegers(seeds);
SU<Nc>::HotConfiguration(pRNG4d, Umu); //std::cout << GridLogMessage << "now hot config" << std::endl;
std::cout << GridLogMessage << "hot cfg done." << std::endl; //SU<Nc>::HotConfiguration(pRNG4d, Umu);
//std::cout << GridLogMessage << "hot cfg done." << std::endl;
// Set up the SAME gauge field on every time plane // Set up the SAME gauge field on every time plane
// int Nt = grid4d->gDimensions()[Tdir]; // int Nt = grid4d->gDimensions()[Tdir];
Grid_unquiesce_nodes(); Grid_unquiesce_nodes();
auto Usft = Umu; Umu_smear = Umu;
Lattice<iScalar<vInteger> > coor(gridHD); Lattice<iScalar<vInteger> > coor(gridHD);
LatticeCoordinate(coor,Tdir); LatticeCoordinate(coor,Tdir);
for(int t=1;t<Nt;t++){ for(int t=1;t<Nt;t++){
// t=1 // t=1
// Umu Usft // Umu Umu_smear
// 0,1,2,3,4,5,6,7 -> 7,0,1,2,3,4,5,6 t=1 // 0,1,2,3,4,5,6,7 -> 7,0,1,2,3,4,5,6 t=1
// 0,0,2,3,4,5,6,7 6,7,0,1,2,3,4,5 t=2 // 0,0,2,3,4,5,6,7 6,7,0,1,2,3,4,5 t=2
// 0,0,0,3,4,5,6,7 5,6,7,0,1,2,3,4 t=3 // 0,0,0,3,4,5,6,7 5,6,7,0,1,2,3,4 t=3
//... //...
Usft = Cshift(Usft,Tdir,-1); Umu_smear = Cshift(Umu_smear,Tdir,-1);
Umu = where(coor==t,Usft,Umu); Umu = where(coor==t,Umu_smear,Umu);
} }
// std::cout << "Umu is "<<Umu<<std::endl; // std::cout << "Umu is "<<Umu<<std::endl;
} }
// Stout smearing // Stout smearing
envGetTmp(GaugeField, Umu_smear);
Umu_smear = Umu; Umu_smear = Umu;
LOG(Message) << "Initial plaquette: " << WilsonLoops<PeriodicGimplR>::avgPlaquette(Umu) << std::endl; LOG(Message) << "Initial plaquette: " << WilsonLoops<PeriodicGimplR>::avgPlaquette(Umu) << std::endl;
{ {
const StoutParameters &Stout{par().Stout};
envGetTmp(GaugeField, Umu_stout); envGetTmp(GaugeField, Umu_stout);
const int &Steps{par().Stout.steps}; Smear_Stout<PeriodicGimplR> LS(Stout.parm);
Smear_Stout<PeriodicGimplR> LS(par().Stout.parm); for (int i = 0; i < Stout.steps; i++) {
for (int i = 0; i < Steps; i++) {
LS.smear(Umu_stout, Umu_smear); LS.smear(Umu_stout, Umu_smear);
Umu_smear = Umu_stout; Umu_smear = Umu_stout;
} }
@ -314,10 +314,9 @@ void TLapEvec<FImpl>::execute(void)
// Invert Peardon Nabla operator separately on each time-slice // Invert Peardon Nabla operator separately on each time-slice
//////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////
std::string sEigenPackName(par().EigenPackName); std::string sEigenPackName(getName());
bool bReturnValue = true; bool bReturnValue = true;
auto & eig4d = envGet(DistilEP, getName() ); auto & eig4d = envGet(DistilEP, getName() );
eig4d.resize(nvec,gridHD);
envGetTmp(std::vector<DistilEP>, eig); // Eigenpack for each timeslice envGetTmp(std::vector<DistilEP>, eig); // Eigenpack for each timeslice
envGetTmp(LatticeGaugeField, UmuNoTime); // Gauge field without time dimension envGetTmp(LatticeGaugeField, UmuNoTime); // Gauge field without time dimension
envGetTmp(LatticeColourVector, src); envGetTmp(LatticeColourVector, src);
@ -330,9 +329,7 @@ void TLapEvec<FImpl>::execute(void)
std::cout << GridLogMessage << " Compute eigenpack, Timeslice = " << t << std::endl; std::cout << GridLogMessage << " Compute eigenpack, Timeslice = " << t << std::endl;
std::cout << GridLogMessage << "------------------------------------------------------------" << std::endl; std::cout << GridLogMessage << "------------------------------------------------------------" << std::endl;
LOG(Message) << "eig.size()=" << eig.size() << std::endl;
eig[t].resize(LPar.Nk+LPar.Np,gridLD); eig[t].resize(LPar.Nk+LPar.Np,gridLD);
LOG(Message) << "After eig[t].resize" << std::endl;
// Construct smearing operator // Construct smearing operator
ExtractSliceLocal(UmuNoTime,Umu_smear,0,t-Ntfirst,Grid::QCD::Tdir); // switch to 3d/4d objects ExtractSliceLocal(UmuNoTime,Umu_smear,0,t-Ntfirst,Grid::QCD::Tdir); // switch to 3d/4d objects

267
tests/hadrons/Test_hadrons_distil.cc
View File

@ -34,19 +34,11 @@ using namespace Grid;
using namespace Hadrons; using namespace Hadrons;
///////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////
// This is copied from the free propagator test // Test creation of laplacian eigenvectors
// Just used as an example - will be deleted
///////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////
void free_prop(Application &application) void test_Global(Application &application)
{ {
std::vector<std::string> flavour = {"h"}; //{"l", "s", "c1", "c2", "c3"};
std::vector<double> mass = {.2}; //{.01, .04, .2 , .25 , .3 };
std::vector<std::string> lepton_flavour = {"mu"};
std::vector<double> lepton_mass = {.2};
unsigned int nt = GridDefaultLatt()[Tp];
// global parameters // global parameters
Application::GlobalPar globalPar; Application::GlobalPar globalPar;
globalPar.trajCounter.start = 1500; globalPar.trajCounter.start = 1500;
@ -54,168 +46,6 @@ void free_prop(Application &application)
globalPar.trajCounter.step = 20; globalPar.trajCounter.step = 20;
globalPar.runId = "test"; globalPar.runId = "test";
application.setPar(globalPar); application.setPar(globalPar);
// gauge field
application.createModule<MGauge::Unit>("gauge");
// unit gauge field for lepton
application.createModule<MGauge::Unit>("free_gauge");
// pt source
MSource::Point::Par ptPar;
ptPar.position = "0 0 0 0";
application.createModule<MSource::Point>("pt", ptPar);
// sink
MSink::Point::Par sinkPar;
sinkPar.mom = "0 0 0";
application.createModule<MSink::ScalarPoint>("sink", sinkPar);
// set fermion boundary conditions to be periodic space, antiperiodic time.
std::string boundary = "1 1 1 -1";
//Propagators from FFT and Feynman rules
for (unsigned int i = 0; i < lepton_mass.size(); ++i)
{
//DWF actions
MAction::DWF::Par actionPar_lep;
actionPar_lep.gauge = "free_gauge";
actionPar_lep.Ls = 8;
actionPar_lep.M5 = 1.8;
actionPar_lep.mass = lepton_mass[i];
actionPar_lep.boundary = boundary;
application.createModule<MAction::DWF>("free_DWF_" + lepton_flavour[i], actionPar_lep);
//DWF free propagators
MFermion::FreeProp::Par freePar;
freePar.source = "pt";
freePar.action = "free_DWF_" + lepton_flavour[i];
freePar.twist = "0 0 0 0.5";
freePar.mass = lepton_mass[i];
application.createModule<MFermion::FreeProp>("Lpt_" + lepton_flavour[i],
freePar);
//Wilson actions
MAction::Wilson::Par actionPar_lep_W;
actionPar_lep_W.gauge = "free_gauge";
actionPar_lep_W.mass = lepton_mass[i];
actionPar_lep_W.boundary = boundary;
application.createModule<MAction::Wilson>("free_W_" + lepton_flavour[i], actionPar_lep_W);
//Wilson free propagators
MFermion::FreeProp::Par freePar_W;
freePar_W.source = "pt";
freePar_W.action = "free_W_" + lepton_flavour[i];
freePar_W.twist = "0 0 0 0.5";
freePar_W.mass = lepton_mass[i];
application.createModule<MFermion::FreeProp>("W_Lpt_" + lepton_flavour[i],
freePar_W);
}
//Propagators from inversion
for (unsigned int i = 0; i < flavour.size(); ++i)
{
//DWF actions
MAction::DWF::Par actionPar;
actionPar.gauge = "gauge";
actionPar.Ls = 8;
actionPar.M5 = 1.8;
actionPar.mass = mass[i];
actionPar.boundary = boundary;
application.createModule<MAction::DWF>("DWF_" + flavour[i], actionPar);
// solvers
MSolver::RBPrecCG::Par solverPar;
solverPar.action = "DWF_" + flavour[i];
solverPar.residual = 1.0e-8;
solverPar.maxIteration = 10000;
application.createModule<MSolver::RBPrecCG>("CG_" + flavour[i],
solverPar);
//DWF propagators
MFermion::GaugeProp::Par quarkPar;
quarkPar.solver = "CG_" + flavour[i];
quarkPar.source = "pt";
application.createModule<MFermion::GaugeProp>("Qpt_" + flavour[i],
quarkPar);
//Wilson actions
MAction::Wilson::Par actionPar_W;
actionPar_W.gauge = "gauge";
actionPar_W.mass = mass[i];
actionPar_W.boundary = boundary;
application.createModule<MAction::Wilson>("W_" + flavour[i], actionPar_W);
// solvers
MSolver::RBPrecCG::Par solverPar_W;
solverPar_W.action = "W_" + flavour[i];
solverPar_W.residual = 1.0e-8;
solverPar_W.maxIteration = 10000;
application.createModule<MSolver::RBPrecCG>("W_CG_" + flavour[i],
solverPar_W);
//Wilson propagators
MFermion::GaugeProp::Par quarkPar_W;
quarkPar_W.solver = "W_CG_" + flavour[i];
quarkPar_W.source = "pt";
application.createModule<MFermion::GaugeProp>("W_Qpt_" + flavour[i],
quarkPar_W);
}
//2pt contraction for Propagators from FFT and Feynman rules
for (unsigned int i = 0; i < lepton_flavour.size(); ++i)
for (unsigned int j = i; j < lepton_flavour.size(); ++j)
{
//2pt function contraction DWF
MContraction::Meson::Par freemesPar;
freemesPar.output = "2pt_free/DWF_L_pt_" + lepton_flavour[i] + lepton_flavour[j];
freemesPar.q1 = "Lpt_" + lepton_flavour[i];
freemesPar.q2 = "Lpt_" + lepton_flavour[j];
freemesPar.gammas = "(Gamma5 Gamma5)";
freemesPar.sink = "sink";
application.createModule<MContraction::Meson>("meson_L_pt_"
+ lepton_flavour[i] + lepton_flavour[j],
freemesPar);
//2pt function contraction Wilson
MContraction::Meson::Par freemesPar_W;
freemesPar_W.output = "2pt_free/W_L_pt_" + lepton_flavour[i] + lepton_flavour[j];
freemesPar_W.q1 = "W_Lpt_" + lepton_flavour[i];
freemesPar_W.q2 = "W_Lpt_" + lepton_flavour[j];
freemesPar_W.gammas = "(Gamma5 Gamma5)";
freemesPar_W.sink = "sink";
application.createModule<MContraction::Meson>("W_meson_L_pt_"
+ lepton_flavour[i] + lepton_flavour[j],
freemesPar_W);
}
//2pt contraction for Propagators from inverion
for (unsigned int i = 0; i < flavour.size(); ++i)
for (unsigned int j = i; j < flavour.size(); ++j)
{
//2pt function contraction DWF
MContraction::Meson::Par mesPar;
mesPar.output = "2pt_free/DWF_pt_" + flavour[i] + flavour[j];
mesPar.q1 = "Qpt_" + flavour[i];
mesPar.q2 = "Qpt_" + flavour[j];
mesPar.gammas = "(Gamma5 Gamma5)";
mesPar.sink = "sink";
application.createModule<MContraction::Meson>("meson_pt_"
+ flavour[i] + flavour[j],
mesPar);
//2pt function contraction Wilson
MContraction::Meson::Par mesPar_W;
mesPar_W.output = "2pt_free/W_pt_" + flavour[i] + flavour[j];
mesPar_W.q1 = "W_Qpt_" + flavour[i];
mesPar_W.q2 = "W_Qpt_" + flavour[j];
mesPar_W.gammas = "(Gamma5 Gamma5)";
mesPar_W.sink = "sink";
application.createModule<MContraction::Meson>("W_meson_pt_"
+ flavour[i] + flavour[j],
mesPar_W);
}
} }
///////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////
@ -225,23 +55,16 @@ void free_prop(Application &application)
void test_LapEvec(Application &application) void test_LapEvec(Application &application)
{ {
const char szGaugeName[] = "gauge"; const char szGaugeName[] = "gauge";
// global parameters
Application::GlobalPar globalPar;
globalPar.trajCounter.start = 1500;
globalPar.trajCounter.end = 1520;
globalPar.trajCounter.step = 20;
globalPar.runId = "test";
application.setPar(globalPar);
// gauge field // gauge field
application.createModule<MGauge::Random>(szGaugeName); application.createModule<MGauge::Random>(szGaugeName);
// Now make an instance of the LapEvec object // Now make an instance of the LapEvec object
MDistil::LapEvecPar p; MDistil::LapEvecPar p;
p.gauge = szGaugeName; p.gauge = szGaugeName;
//p.EigenPackName = "ePack"; //p.EigenPackName = "ePack";
p.Distil.TI = 8; //p.Distil.TI = 8;
p.Distil.LI = 3; //p.Distil.LI = 3;
p.Distil.Nnoise = 2; //p.Distil.Nnoise = 2;
p.Distil.tSrc = 0; //p.Distil.tSrc = 0;
p.Stout.steps = 3; p.Stout.steps = 3;
p.Stout.parm = 0.2; p.Stout.parm = 0.2;
p.Cheby.PolyOrder = 11; p.Cheby.PolyOrder = 11;
@ -261,60 +84,9 @@ void test_LapEvec(Application &application)
void test_Perambulators(Application &application) void test_Perambulators(Application &application)
{ {
const unsigned int nt = GridDefaultLatt()[Tp];
// global parameters
Application::GlobalPar globalPar;
globalPar.trajCounter.start = 3000;
globalPar.trajCounter.end = 3040;
globalPar.trajCounter.step = 40;
globalPar.runId = "test";
application.setPar(globalPar);
// gauge field
application.createModule<MGauge::Unit>("gauge");
// Now make an instance of the LapEvec object
application.createModule<MDistil::PerambLight>("PerambulatorsInstance");
}
/////////////////////////////////////////////////////////////
// DistilVectors
/////////////////////////////////////////////////////////////
void test_DistilVectors(Application &application)
{
const unsigned int nt = GridDefaultLatt()[Tp];
const char szGaugeName[] = "gauge";
// global parameters
Application::GlobalPar globalPar;
globalPar.trajCounter.start = 1500;
globalPar.trajCounter.end = 1520;
globalPar.trajCounter.step = 20;
globalPar.runId = "test";
application.setPar(globalPar);
// gauge field
application.createModule<MGauge::Random>(szGaugeName);
// Now make an instance of the LapEvec object
MDistil::LapEvecPar p;
p.gauge = szGaugeName;
//p.EigenPackName = "eigenPack";
p.Distil.TI = 8;
p.Distil.LI = 3;
p.Distil.Nnoise = 2;
p.Distil.tSrc = 0;
p.Stout.steps = 3;
p.Stout.parm = 0.2;
p.Cheby.PolyOrder = 11;
p.Cheby.alpha = 0.3;
p.Cheby.beta = 12.5;
p.Lanczos.Nvec = 5;
p.Lanczos.Nk = 6;
p.Lanczos.Np = 2;
p.Lanczos.MaxIt = 1000;
p.Lanczos.resid = 1e-2;
application.createModule<MDistil::LapEvec>("LapEvec",p);
// PerambLight parameters // PerambLight parameters
MDistil::PerambLight::Par PerambPar; MDistil::PerambLight::Par PerambPar;
PerambPar.eigenPack="LapEvec_eigenPack"; PerambPar.eigenPack="LapEvec";
PerambPar.tsrc = 0; PerambPar.tsrc = 0;
PerambPar.nnoise = 1; PerambPar.nnoise = 1;
PerambPar.LI=6; PerambPar.LI=6;
@ -330,6 +102,13 @@ void test_DistilVectors(Application &application)
PerambPar.CGPrecision=1e-8; PerambPar.CGPrecision=1e-8;
PerambPar.MaxIterations=10000; PerambPar.MaxIterations=10000;
application.createModule<MDistil::PerambLight>("Peramb",PerambPar); application.createModule<MDistil::PerambLight>("Peramb",PerambPar);
}
/////////////////////////////////////////////////////////////
// DistilVectors
/////////////////////////////////////////////////////////////
void test_DistilVectors(Application &application)
{
// DistilVectors parameters // DistilVectors parameters
MDistil::DistilVectors::Par DistilVecPar; MDistil::DistilVectors::Par DistilVecPar;
DistilVecPar.noise="Peramb_noise"; DistilVecPar.noise="Peramb_noise";
@ -421,16 +200,22 @@ int main(int argc, char *argv[])
// For now perform free propagator test - replace this with distillation test(s) // For now perform free propagator test - replace this with distillation test(s)
LOG(Message) << "====== Creating xml for test " << iTestNum << " ======" << std::endl; LOG(Message) << "====== Creating xml for test " << iTestNum << " ======" << std::endl;
const unsigned int nt = GridDefaultLatt()[Tp]; //const unsigned int nt = GridDefaultLatt()[Tp];
switch(iTestNum) { switch(iTestNum) {
case 0:
free_prop( application );
break;
case 1: case 1:
test_Global( application );
test_LapEvec( application ); test_LapEvec( application );
break; break;
default: // 2 case 2:
test_Global( application );
test_LapEvec( application );
test_Perambulators( application );
break;
default: // 3
test_Global( application );
test_LapEvec( application );
test_Perambulators( application );
test_DistilVectors( application ); test_DistilVectors( application );
break; break;
} }