mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-10 07:55:35 +00:00
70 lines
2.0 KiB
C++
70 lines
2.0 KiB
C++
#include "Grid.h"
|
|
|
|
|
|
using namespace std;
|
|
using namespace Grid;
|
|
using namespace Grid::QCD;
|
|
|
|
int main (int argc, char ** argv)
|
|
{
|
|
Grid_init(&argc,&argv);
|
|
|
|
std::vector<int> latt_size = GridDefaultLatt();
|
|
std::vector<int> simd_layout = GridDefaultSimd(4,vComplex::Nsimd());
|
|
std::vector<int> mpi_layout = GridDefaultMpi();
|
|
|
|
GridCartesian Fine(latt_size,simd_layout,mpi_layout);
|
|
GridRedBlackCartesian RBFine(latt_size,simd_layout,mpi_layout);
|
|
|
|
std::vector<int> seeds({6,7,8,80});
|
|
GridParallelRNG pRNG(&Fine);
|
|
pRNG.SeedFixedIntegers(seeds);
|
|
|
|
std::vector<int> seedsS({1,2,3,4});
|
|
GridSerialRNG sRNG;
|
|
sRNG.SeedFixedIntegers(seedsS);
|
|
|
|
LatticeLorentzColourMatrix U(&Fine);
|
|
|
|
SU3::HotConfiguration(pRNG, U);
|
|
|
|
// simplify template declaration? Strip the lorentz from the second template
|
|
WilsonGaugeActionR Waction(5.6);
|
|
|
|
Real mass=-0.77;
|
|
WilsonFermionR FermOp(U,Fine,RBFine,mass);
|
|
|
|
ConjugateGradient<LatticeFermion> CG(1.0e-8,10000);
|
|
|
|
TwoFlavourEvenOddPseudoFermionAction<WilsonImplR> WilsonNf2(FermOp,CG,CG);
|
|
|
|
//Collect actions
|
|
ActionLevel<LatticeGaugeField> Level1(1);
|
|
ActionLevel<LatticeGaugeField> Level2(4);
|
|
Level1.push_back(&WilsonNf2);
|
|
Level2.push_back(&Waction);
|
|
|
|
ActionSet<LatticeGaugeField> FullSet;
|
|
FullSet.push_back(Level1);
|
|
FullSet.push_back(Level2);
|
|
|
|
// Create integrator
|
|
//typedef LeapFrog<LatticeGaugeField> IntegratorAlgorithm;// change here to change the algorithm
|
|
//typedef MinimumNorm2<LatticeGaugeField> IntegratorAlgorithm;// change here to change the algorithm
|
|
typedef ForceGradient<LatticeGaugeField> IntegratorAlgorithm;// change here to change the algorithm
|
|
|
|
IntegratorParameters MDpar(16);
|
|
IntegratorAlgorithm MDynamics(&Fine,MDpar, FullSet);
|
|
|
|
|
|
// Create HMC
|
|
NerscHmcCheckpointer<LatticeGaugeField> Checkpoint(std::string("ckpoint_lat"),std::string("ckpoint_rng"),1);
|
|
HMCparameters HMCpar;
|
|
HybridMonteCarlo<LatticeGaugeField,IntegratorAlgorithm> HMC(HMCpar, MDynamics,sRNG,pRNG,U);
|
|
HMC.AddObservable(&Checkpoint);
|
|
|
|
// Run it
|
|
HMC.evolve();
|
|
|
|
}
|