mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-10 07:55:35 +00:00
142 lines
4.5 KiB
C++
142 lines
4.5 KiB
C++
/*************************************************************************************
|
|
|
|
Grid physics library, www.github.com/paboyle/Grid
|
|
|
|
Source file: ./tests/Test_hmc_WilsonFermionGauge.cc
|
|
|
|
Copyright (C) 2015
|
|
|
|
Author: Peter Boyle <pabobyle@ph.ed.ac.uk>
|
|
Author: neo <cossu@post.kek.jp>
|
|
Author: Guido Cossu <guido.cossu@ed.ac.uk>
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with this program; if not, write to the Free Software Foundation, Inc.,
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
|
|
See the full license in the file "LICENSE" in the top level distribution
|
|
directory
|
|
*************************************************************************************/
|
|
/* END LEGAL */
|
|
#include <Grid/Grid.h>
|
|
|
|
using namespace std;
|
|
using namespace Grid;
|
|
using namespace Grid::QCD;
|
|
|
|
namespace Grid {
|
|
namespace QCD {
|
|
|
|
//Change here the type of reader
|
|
typedef Grid::TextReader InputFileReader;
|
|
|
|
|
|
class HMCRunnerParameters : Serializable {
|
|
public:
|
|
GRID_SERIALIZABLE_CLASS_MEMBERS(HMCRunnerParameters,
|
|
double, beta,
|
|
int, MDsteps,
|
|
double, TrajectorLength,
|
|
int, SaveInterval,
|
|
std::string, format,
|
|
std::string, conf_prefix,
|
|
std::string, rng_prefix,
|
|
std::string, serial_seeds,
|
|
std::string, parallel_seeds,
|
|
);
|
|
|
|
HMCRunnerParameters() {}
|
|
};
|
|
|
|
|
|
// Derive from the BinaryHmcRunner (templated for gauge fields)
|
|
class HmcRunner : public BinaryHmcRunner {
|
|
public:
|
|
HMCRunnerParameters HMCPar;
|
|
void BuildTheAction(int argc, char **argv)
|
|
|
|
{
|
|
// Typedefs to simplify notation
|
|
typedef WilsonGaugeActionR GaugeAction;
|
|
typedef WilsonImplR ImplPolicy;
|
|
typedef WilsonFermionR FermionAction;
|
|
typedef typename FermionAction::FermionField FermionField;
|
|
|
|
// this can be simplified too. MakeDefaultGrid(Nd)
|
|
UGrid = SpaceTimeGrid::makeFourDimGrid(
|
|
GridDefaultLatt(),
|
|
GridDefaultSimd(Nd, vComplex::Nsimd()),
|
|
GridDefaultMpi());
|
|
|
|
|
|
// Gauge action
|
|
GaugeAction Waction(HMCPar.beta);
|
|
|
|
// Collect actions
|
|
ActionLevel<Field> Level1(1);
|
|
Level1.push_back(&Waction);
|
|
TheAction.push_back(Level1);
|
|
|
|
// Add observables
|
|
// options for checkpointers
|
|
//BinaryHmcCheckpointer
|
|
//ILDGHmcCheckpointer
|
|
//NerscHmcCheckpointer
|
|
NerscHmcCheckpointer<BinaryHmcRunner::ImplPolicy> Checkpoint(
|
|
HMCPar.conf_prefix, HMCPar.rng_prefix, HMCPar.SaveInterval, HMCPar.format);
|
|
// Can implement also a specific function in the hmcrunner
|
|
// AddCheckpoint (...) that takes the same parameters + a string/tag
|
|
// defining the type of the checkpointer
|
|
// with tags can be implemented by overloading and no ifs
|
|
// Then force all checkpoint to have few common functions
|
|
// return an object that is then passed to the Run function
|
|
|
|
PlaquetteLogger<BinaryHmcRunner::ImplPolicy> PlaqLog(
|
|
std::string("Plaquette"));
|
|
ObservablesList.push_back(&PlaqLog);
|
|
ObservablesList.push_back(&Checkpoint);
|
|
|
|
Run(argc, argv, Checkpoint); // no smearing
|
|
};
|
|
};
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
Grid_init(&argc, &argv);
|
|
|
|
int threads = GridThread::GetThreads();
|
|
std::cout << GridLogMessage << "Grid is setup to use " << threads
|
|
<< " threads" << std::endl;
|
|
|
|
HmcRunner TheHMC;
|
|
// make input file name general
|
|
InputFileReader Reader("input.wilson_gauge.params");
|
|
read(Reader, "HMC", TheHMC.HMCPar);
|
|
|
|
std::cout << GridLogMessage << TheHMC.HMCPar << std::endl;
|
|
|
|
// Seeds for the random number generators
|
|
// generalise
|
|
std::vector<int> SerSeed = strToVec<int>(TheHMC.HMCPar.serial_seeds);
|
|
std::vector<int> ParSeed = strToVec<int>(TheHMC.HMCPar.parallel_seeds);
|
|
|
|
TheHMC.RNGSeeds(SerSeed, ParSeed);
|
|
|
|
TheHMC.MDparameters.set(TheHMC.HMCPar.MDsteps, TheHMC.HMCPar.TrajectorLength);
|
|
|
|
TheHMC.BuildTheAction(argc, argv);
|
|
|
|
Grid_finalize();
|
|
}
|