/************************************************************************************* Grid physics library, www.github.com/paboyle/Grid Source file: ./tests/Test_padded_cell.cc Copyright (C) 2023 Author: Peter Boyle 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 */ // copied here from Test_general_coarse_pvdagm.cc #include #include #include #include #include #include #include using namespace std; using namespace Grid; namespace Grid { struct LanczosParameters: Serializable { GRID_SERIALIZABLE_CLASS_MEMBERS(LanczosParameters, RealD, mass , RealD, mstep , Integer, Nstop, Integer, Nk, Integer, Np, Integer, ReadEvec, RealD, resid, RealD, ChebyLow, RealD, ChebyHigh, Integer, ChebyOrder) LanczosParameters() { ////////////////////////////// Default values mass = 0; ///////////////////////////////// } template LanczosParameters(Reader & TheReader){ initialize(TheReader); } template < class ReaderClass > void initialize(Reader &TheReader){ // std::cout << GridLogMessage << "Reading HMC\n"; read(TheReader, "HMC", *this); } void print_parameters() const { // std::cout << GridLogMessage << "[HMC parameters] Trajectories : " << Trajectories << "\n"; // std::cout << GridLogMessage << "[HMC parameters] Start trajectory : " << StartTrajectory << "\n"; // std::cout << GridLogMessage << "[HMC parameters] Metropolis test (on/off): " << std::boolalpha << MetropolisTest << "\n"; // std::cout << GridLogMessage << "[HMC parameters] Thermalization trajs : " << NoMetropolisUntil << "\n"; // std::cout << GridLogMessage << "[HMC parameters] Starting type : " << StartingType << "\n"; // MD.print_parameters(); } }; } template void writeFile(T& in, std::string const fname){ #if 1 // Ref: https://github.com/paboyle/Grid/blob/feature/scidac-wp1/tests/debug/Test_general_coarse_hdcg_phys48.cc#L111 std::cout << Grid::GridLogMessage << "Writes to: " << fname << std::endl; Grid::emptyUserRecord record; Grid::ScidacWriter WR(in.Grid()->IsBoss()); WR.open(fname); WR.writeScidacFieldRecord(in,record,0); WR.close(); #endif // What is the appropriate way to throw error? } typedef WilsonFermionD WilsonOp; typedef typename WilsonFermionD::FermionField FermionField; #if 0 // Hermitize a DWF operator by squaring it template class SquaredLinearOperator : public LinearOperatorBase { public: Matrix &_Mat; public: SquaredLinearOperator(Matrix &Mat): _Mat(Mat) {}; void OpDiag (const Field &in, Field &out) { assert(0); } void OpDir (const Field &in, Field &out,int dir,int disp) { assert(0); } void OpDirAll (const Field &in, std::vector &out){ assert(0); }; void Op (const Field &in, Field &out){ // std::cout << "Op is overloaded as HermOp" << std::endl; HermOp(in, out); } void AdjOp (const Field &in, Field &out){ HermOp(in, out); } void _Op (const Field &in, Field &out){ // std::cout << "Op: M "< class PVdagMLinearOperator : public LinearOperatorBase { Matrix &_Mat; Matrix &_PV; public: PVdagMLinearOperator(Matrix &Mat,Matrix &PV): _Mat(Mat),_PV(PV){}; void OpDiag (const Field &in, Field &out) { assert(0); } void OpDir (const Field &in, Field &out,int dir,int disp) { assert(0); } void OpDirAll (const Field &in, std::vector &out){ assert(0); }; void Op (const Field &in, Field &out){ std::cout << "Op: PVdag M "< class ShiftedPVdagMLinearOperator : public LinearOperatorBase { Matrix &_Mat; Matrix &_PV; RealD shift; public: ShiftedPVdagMLinearOperator(RealD _shift,Matrix &Mat,Matrix &PV): shift(_shift),_Mat(Mat),_PV(PV){}; void OpDiag (const Field &in, Field &out) { assert(0); } void OpDir (const Field &in, Field &out,int dir,int disp) { assert(0); } void OpDirAll (const Field &in, std::vector &out){ assert(0); }; void Op (const Field &in, Field &out){ std::cout << "Op: PVdag M "< class ShiftedComplexPVdagMLinearOperator : public LinearOperatorBase { Matrix &_Mat; Matrix &_PV; ComplexD shift; public: ShiftedComplexPVdagMLinearOperator(ComplexD _shift,Matrix &Mat,Matrix &PV): shift(_shift),_Mat(Mat),_PV(PV){}; void OpDiag (const Field &in, Field &out) { assert(0); } void OpDir (const Field &in, Field &out,int dir,int disp) { assert(0); } void OpDirAll (const Field &in, std::vector &out){ assert(0); }; void Op (const Field &in, Field &out){ std::cout << "Op: PVdag M "< class MGPreconditioner : public LinearFunction< Lattice > { public: using LinearFunction >::operator(); typedef Aggregation Aggregates; typedef typename Aggregation::FineField FineField; typedef typename Aggregation::CoarseVector CoarseVector; typedef typename Aggregation::CoarseMatrix CoarseMatrix; typedef LinearOperatorBase FineOperator; typedef LinearFunction FineSmoother; typedef LinearOperatorBase CoarseOperator; typedef LinearFunction CoarseSolver; Aggregates & _Aggregates; FineOperator & _FineOperator; FineSmoother & _PreSmoother; FineSmoother & _PostSmoother; CoarseOperator & _CoarseOperator; CoarseSolver & _CoarseSolve; int level; void Level(int lv) {level = lv; }; MGPreconditioner(Aggregates &Agg, FineOperator &Fine, FineSmoother &PreSmoother, FineSmoother &PostSmoother, CoarseOperator &CoarseOperator_, CoarseSolver &CoarseSolve_) : _Aggregates(Agg), _FineOperator(Fine), _PreSmoother(PreSmoother), _PostSmoother(PostSmoother), _CoarseOperator(CoarseOperator_), _CoarseSolve(CoarseSolve_), level(1) { } virtual void operator()(const FineField &in, FineField & out) { GridBase *CoarseGrid = _Aggregates.CoarseGrid; // auto CoarseGrid = _CoarseOperator.Grid(); CoarseVector Csrc(CoarseGrid); CoarseVector Csol(CoarseGrid); FineField vec1(in.Grid()); FineField vec2(in.Grid()); std::cout< void testSchurFromHess(Arnoldi& Arn, Field& src, int Nlarge, int Nm, int Nk) { std::cout< lat_size {32, 32, 32, 32}; // std::cout << "Lattice size: " << lat_size << std::endl; GridCartesian * UGrid = SpaceTimeGrid::makeFourDimGrid(GridDefaultLatt(), GridDefaultSimd(Nd,vComplex::Nsimd()), GridDefaultMpi()); GridRedBlackCartesian * UrbGrid = SpaceTimeGrid::makeFourDimRedBlackGrid(UGrid); // GridCartesian * FGrid = SpaceTimeGrid::makeFiveDimGrid(Ls,UGrid); // GridRedBlackCartesian * FrbGrid = SpaceTimeGrid::makeFiveDimRedBlackGrid(Ls,UGrid); GridCartesian * FGrid = UGrid; GridRedBlackCartesian * FrbGrid = UrbGrid; // Construct a coarsened grid // poare TODO: replace this with the following line? Coordinate clatt = GridDefaultLatt(); // Coordinate clatt = GridDefaultLatt(); // [PO] initial line before I edited it for(int d=0;d seeds4({1,2,3,4}); std::vector seeds5({5,6,7,8}); std::vector cseeds({5,6,7,8}); GridParallelRNG RNG5(FGrid); RNG5.SeedFixedIntegers(seeds5); GridParallelRNG RNG4(UGrid); RNG4.SeedFixedIntegers(seeds4); GridParallelRNG CRNG(Coarse5d);CRNG.SeedFixedIntegers(cseeds); LatticeFermion result(FGrid); result=Zero(); LatticeFermion ref(FGrid); ref=Zero(); LatticeFermion tmp(FGrid); LatticeFermion err(FGrid); LatticeGaugeField Umu(UGrid); FieldMetaData header; // std::string file("ckpoint_lat.4000"); // std::string file("/Users/patrickoare/libraries/PETSc-Grid/ckpoint_lat.4000"); std::string file("./config"); NerscIO::readConfiguration(Umu,header,file); LanczosParameters LanParams; { XmlReader HMCrd("LanParams.xml"); read(HMCrd,"LanczosParameters",LanParams); } std::cout << GridLogMessage<< LanParams < PM; PM(PVdagM, src); int Nm = 50; int Nk = 12; int Np = 38; // int Nk = Nm+1; // if just running once int maxIter = 10000; int Nstop = 10; RealD resid = 1.0e-5; std::vector boundary = {1,1,1,-1}; WilsonOp::ImplParams Params(boundary); // DomainWallFermionD Ddwf(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,mass,M5); // DomainWallFermionD Dpv(Umu,*FGrid,*FrbGrid,*UGrid,*UrbGrid,1.0,M5); mass=LanParams.mass; std::cout << GridLogIRL<< "mass "< LittleDiracOperator; typedef LittleDiracOperator::CoarseVector CoarseVector; NextToNearestStencilGeometry5D geom(Coarse5d); std::cout< PVdagM_t; // typedef ShiftedPVdagMLinearOperator ShiftedPVdagM_t; // typedef ShiftedComplexPVdagMLinearOperator ShiftedComplexPVdagM_t; // PVdagM_t PVdagM(Ddwf, Dpv); // ShiftedPVdagM_t ShiftedPVdagM(0.1,Ddwf,Dpv); // SquaredLinearOperator Dsq (Ddwf); // NonHermitianLinearOperator DLinOp (Ddwf); NonHermitianLinearOperator Dwilson(WilsonOperator); /// <----- MdagMLinearOperator HermOp(WilsonOperator); /// <----- Gamma5HermitianLinearOperator HermOp2(WilsonOperator); /// <---- // PowerMethod PM; PM(PVdagM, src); resid=LanParams.resid; Nstop=LanParams.Nstop; Nk=LanParams.Nk; Np=LanParams.Np; Nm = Nk + Np; int Nu=16; std::vector src(Nu,FGrid); for(int i=0;i(Arn, src, 10, 6, 4); Arnoldi Arn2 (DLinOp, FGrid, 1e-8); testSchurFromHess(Arn2, src, 16, 12, 8); */ std::cout<