mirror of
https://github.com/paboyle/Grid.git
synced 2025-04-04 03:05:55 +01:00
fd933420c6
Added a bounds-check function for the RHMC with arbitrary power
Added a pseudofermion action for the rational ratio with an arbitrary power and a mixed-precision variant of the same. The existing one-flavor rational ratio class now uses the general class under the hood
To support testing of the two-flavor even-odd ratio pseudofermion, separated the functionality of generating the random field and performing the heatbath step, and added a method to obtain the pseudofermion field
Added a new HMC runner start type: CheckpointStartReseed, which reseeds the RNG from scratch, allowing for the creation of new evolution streams from an existing checkpoint. Added log output of seeds used when the RNG is seeded.
EOFA changes:
To support mixed-precision inversion, generalized the class to maintain a separate solver for the L and R operators in the heatbath (separate solvers are already implemented for the other stages)
To support mixed-precision, the action of setting the operator shift coefficients is now maintained in a virtual function. A derived class for mixed-precision solvers ensures the coefficients are applied to both the double and single-prec operators
The ||^2 of the random source is now stored by the heatbath and compared to the initial action when it is computed. These should be equal but may differ if the rational bounds are not chosen correctly, hence serving as a useful and free test
Fixed calculation of M_eofa (previously incomplete and #if'd out)
Added functionality to compute M_eofa^-1 to complement the calculation of M_eofa (both are equally expensive!)
To support testing, separated the functionality of generating the random field and performing the heatbath step, and added a method to obtain the pseudofermion field
Added a test program which computes the G-parity force using the 1 and 2 flavor implementations and compares the result. Test supports DWF, EOFA and DSDR actions, chosen by a command line option.
The Mobius EOFA force test now also checks the rational approximation used for the heatbath
Added a test program for the mixed precision EOFA compared to the double-prec implementation,
G-parity HMC test now applied GPBC in the y direction and not the t direction (GPBC in t are no longer supported) and checkpoints after every configuration
Added a test program which computes the two-flavor G-parity action (via RHMC) with both the 1 and 2 flavor implementations and checks they agree
Added a test program to check the implementation of M_eofa^{-1}
109 lines
3.2 KiB
C++
109 lines
3.2 KiB
C++
/*************************************************************************************
|
|
|
|
Grid physics library, www.github.com/paboyle/Grid
|
|
|
|
Source file: ./lib/qcd/hmc/GenericHmcRunner.h
|
|
|
|
Copyright (C) 2015
|
|
Copyright (C) 2016
|
|
|
|
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 */
|
|
#ifndef GRID_HMC_MODULES
|
|
#define GRID_HMC_MODULES
|
|
|
|
#include "HMC_GridModules.h"
|
|
|
|
NAMESPACE_BEGIN(Grid);
|
|
|
|
////////////////////////////////////////////////////////////////////
|
|
struct RNGModuleParameters: Serializable {
|
|
GRID_SERIALIZABLE_CLASS_MEMBERS(RNGModuleParameters,
|
|
std::string, serial_seeds,
|
|
std::string, parallel_seeds,);
|
|
|
|
std::vector<int> getSerialSeeds(){return strToVec<int>(serial_seeds);}
|
|
std::vector<int> getParallelSeeds(){return strToVec<int>(parallel_seeds);}
|
|
|
|
RNGModuleParameters(): serial_seeds("1"), parallel_seeds("1"){}
|
|
|
|
template <class ReaderClass >
|
|
RNGModuleParameters(Reader<ReaderClass>& Reader){
|
|
read(Reader, "RandomNumberGenerator", *this);
|
|
}
|
|
|
|
};
|
|
|
|
// Random number generators module
|
|
class RNGModule{
|
|
GridSerialRNG sRNG_;
|
|
std::unique_ptr<GridParallelRNG> pRNG_;
|
|
RNGModuleParameters Params_;
|
|
|
|
public:
|
|
|
|
RNGModule(){};
|
|
|
|
void set_pRNG(GridParallelRNG* pRNG){
|
|
pRNG_.reset(pRNG);
|
|
}
|
|
|
|
void set_RNGSeeds(RNGModuleParameters& Params) {
|
|
Params_ = Params;
|
|
}
|
|
|
|
GridSerialRNG& get_sRNG() { return sRNG_; }
|
|
GridParallelRNG& get_pRNG() { return *pRNG_.get(); }
|
|
|
|
void seed() {
|
|
auto SerialSeeds = Params_.getSerialSeeds();
|
|
auto ParallelSeeds = Params_.getParallelSeeds();
|
|
if (SerialSeeds.size() == 0 && ParallelSeeds.size() == 0) {
|
|
std::cout << GridLogError << "Seeds not initialized" << std::endl;
|
|
exit(1);
|
|
}
|
|
std::cout << GridLogMessage << "Reseeding serial RNG with seed vector " << SerialSeeds << std::endl;
|
|
sRNG_.SeedFixedIntegers(SerialSeeds);
|
|
std::cout << GridLogMessage << "Reseeding parallel RNG with seed vector " << ParallelSeeds << std::endl;
|
|
pRNG_->SeedFixedIntegers(ParallelSeeds);
|
|
}
|
|
};
|
|
|
|
|
|
/*
|
|
///////////////////////////////////////////////////////////////////
|
|
/// Smearing module
|
|
template <class ImplementationPolicy>
|
|
class SmearingModule{
|
|
virtual void get_smearing();
|
|
};
|
|
|
|
template <class ImplementationPolicy>
|
|
class StoutSmearingModule: public SmearingModule<ImplementationPolicy>{
|
|
SmearedConfiguration<ImplementationPolicy> SmearingPolicy;
|
|
};
|
|
|
|
*/
|
|
|
|
NAMESPACE_END(Grid);
|
|
|
|
#endif // GRID_HMC_MODULES
|