mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-14 01:35:36 +00:00
Merge branch 'feature/distil' of github.com:mmphys/Grid into feature/distil
This commit is contained in:
commit
b74492a805
@ -8,60 +8,61 @@
|
|||||||
BEGIN_HADRONS_NAMESPACE
|
BEGIN_HADRONS_NAMESPACE
|
||||||
|
|
||||||
/******************************************************************************
|
/******************************************************************************
|
||||||
* * Class to generate Distillation $\varrho$ and $\varphi$ vectors *
|
* Class to generate Distillation $\varrho$ and $\varphi$ vectors *
|
||||||
* ******************************************************************************/
|
******************************************************************************/
|
||||||
/*
|
/*
|
||||||
* template <typename FImpl>
|
template <typename FImpl>
|
||||||
* class DistilVectorsFromPerambulator
|
class DistilVectorsFromPerambulator
|
||||||
* {
|
{
|
||||||
* public:
|
public:
|
||||||
* FERM_TYPE_ALIASES(FImpl,);
|
FERM_TYPE_ALIASES(FImpl,);
|
||||||
* SOLVER_TYPE_ALIASES(FImpl,);
|
SOLVER_TYPE_ALIASES(FImpl,);
|
||||||
* public:
|
public:
|
||||||
* DistilVectorsFromPerambulator(FMat &action);
|
DistilVectorsFromPerambulator(FMat &action);
|
||||||
* virtual ~DistilVectorsFromPerambulator(void) = default;
|
virtual ~DistilVectorsFromPerambulator(void) = default;
|
||||||
* void makeRho(FermionField &rhoOut,
|
void makeRho(FermionField &rhoOut,
|
||||||
* const FermionField &evec, const Complex &noises);
|
const FermionField &evec, const Complex &noises);
|
||||||
* void makePhi(FermionField &phiOut,
|
void makePhi(FermionField &phiOut,
|
||||||
* const FermionField &evec, const SpinVector &Perambulators);
|
const FermionField &evec, const SpinVector &Perambulators);
|
||||||
* private:
|
private:
|
||||||
* GridBase *fGrid_, *frbGrid_, *gGrid_;
|
GridBase *fGrid_, *frbGrid_, *gGrid_;
|
||||||
* bool is5d_;
|
bool is5d_;
|
||||||
* FermionField src_o_, sol_e_, sol_o_, tmp_, tmp5_;
|
FermionField src_o_, sol_e_, sol_o_, tmp_, tmp5_;
|
||||||
* };
|
};
|
||||||
*
|
|
||||||
* */
|
*/
|
||||||
|
|
||||||
|
|
||||||
/******************************************************************************
|
/******************************************************************************
|
||||||
* * * A2AVectorsSchurDiagTwo template implementation *
|
* DistilVectorsFromPerambulator template implementation *
|
||||||
* * ******************************************************************************/
|
******************************************************************************/
|
||||||
/*
|
/*
|
||||||
*
|
|
||||||
* template <typename FImpl>
|
|
||||||
* A2AVectorsSchurDiagTwo<FImpl>::A2AVectorsSchurDiagTwo(FMat &action)
|
|
||||||
* : action_(action)
|
|
||||||
* , fGrid_(action_.FermionGrid())
|
|
||||||
* , frbGrid_(action_.FermionRedBlackGrid())
|
|
||||||
* , gGrid_(action_.GaugeGrid())
|
|
||||||
* , src_o_(frbGrid_)
|
|
||||||
* , sol_e_(frbGrid_)
|
|
||||||
* , sol_o_(frbGrid_)
|
|
||||||
* , tmp_(frbGrid_)
|
|
||||||
* , tmp5_(fGrid_)
|
|
||||||
* , op_(action_)
|
|
||||||
* {}
|
|
||||||
*
|
|
||||||
* template <typename FImpl>
|
|
||||||
* void A2AVectorsSchurDiagTwo<FImpl>::makeRho(FermionField &rhoOut, const FermionField &evec, const Complex &noises)
|
|
||||||
* {
|
|
||||||
*
|
|
||||||
* LatticeSpinColourVector tmp2(grid4d);
|
|
||||||
* LatticeSpinColourVector tmp3d(grid3d);
|
|
||||||
* LatticeColourVector tmp3d_nospin(grid3d);
|
|
||||||
* LatticeColourVector tmp_nospin(grid4d);
|
|
||||||
*
|
|
||||||
|
|
||||||
|
template <typename FImpl>
|
||||||
|
DistilVectorsFromPerambulator<FImpl>::DistilVectorsFromPerambulator(FMat &action)
|
||||||
|
: action_(action)
|
||||||
|
, fGrid_(action_.FermionGrid())
|
||||||
|
, frbGrid_(action_.FermionRedBlackGrid())
|
||||||
|
, gGrid_(action_.GaugeGrid())
|
||||||
|
, src_o_(frbGrid_)
|
||||||
|
, sol_e_(frbGrid_)
|
||||||
|
, sol_o_(frbGrid_)
|
||||||
|
, tmp_(frbGrid_)
|
||||||
|
, tmp5_(fGrid_)
|
||||||
|
, op_(action_)
|
||||||
|
{}
|
||||||
|
|
||||||
|
template <typename FImpl>
|
||||||
|
void DistilVectorsFromPerambulator<FImpl>::makeRho(FermionField &rhoOut, const FermionField &evec, const Complex &noises)
|
||||||
|
{
|
||||||
|
|
||||||
|
LatticeSpinColourVector tmp2(grid4d);
|
||||||
|
LatticeSpinColourVector tmp3d(grid3d);
|
||||||
|
LatticeColourVector tmp3d_nospin(grid3d);
|
||||||
|
LatticeColourVector tmp_nospin(grid4d);
|
||||||
|
LatticeColourVector evec3d(grid3d);
|
||||||
|
|
||||||
|
int vecindex;
|
||||||
for (int inoise = 0; inoise < nnoise; inoise++) {
|
for (int inoise = 0; inoise < nnoise; inoise++) {
|
||||||
for (int dk = 0; dk < LI; dk++) {
|
for (int dk = 0; dk < LI; dk++) {
|
||||||
for (int dt = 0; dt < Nt_inv; dt++) {
|
for (int dt = 0; dt < Nt_inv; dt++) {
|
||||||
@ -74,7 +75,8 @@ BEGIN_HADRONS_NAMESPACE
|
|||||||
if( it >= Ntfirst && it < Ntfirst + Ntlocal ) {
|
if( it >= Ntfirst && it < Ntfirst + Ntlocal ) {
|
||||||
for (int ik = dk; ik < nvec; ik += LI){
|
for (int ik = dk; ik < nvec; ik += LI){
|
||||||
for (int is = ds; is < Ns; is += Ns){ //at the moment, full spin dilution is enforced
|
for (int is = ds; is < Ns; is += Ns){ //at the moment, full spin dilution is enforced
|
||||||
tmp3d_nospin = eig[it].evec[ik] * noises[inoise][it][ik]()(is)(); //noises do not have to be a spin vector
|
ExtractSliceLocal(evec3d,eig4d.evec[ik],0,it,3);
|
||||||
|
tmp3d_nospin = evec3d * noises[inoise][it][ik]()(is)(); //noises do not have to be a spin vector
|
||||||
tmp3d=zero;
|
tmp3d=zero;
|
||||||
pokeSpin(tmp3d,tmp3d_nospin,is);
|
pokeSpin(tmp3d,tmp3d_nospin,is);
|
||||||
tmp2=zero;
|
tmp2=zero;
|
||||||
@ -96,6 +98,42 @@ BEGIN_HADRONS_NAMESPACE
|
|||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <typename FImpl>
|
||||||
|
void DistilVectorsFromPerambulator<FImpl>::makePhi(FermionField &rhoOut, const FermionField &evec, const SpinVector &perambulator)
|
||||||
|
{
|
||||||
|
|
||||||
|
LatticeSpinColourVector tmp2(grid4d);
|
||||||
|
LatticeSpinColourVector tmp3d(grid3d);
|
||||||
|
LatticeColourVector tmp3d_nospin(grid3d);
|
||||||
|
LatticeColourVector tmp_nospin(grid4d);
|
||||||
|
LatticeColourVector evec3d(grid3d);
|
||||||
|
|
||||||
|
int vecindex;
|
||||||
|
for (int inoise = 0; inoise < nnoise; inoise++) {
|
||||||
|
for (int dk = 0; dk < LI; dk++) {
|
||||||
|
for (int dt = 0; dt < Nt_inv; dt++) {
|
||||||
|
for (int ds = 0; ds < Ns; ds++) {
|
||||||
|
vecindex = inoise + nnoise * dk + nnoise * LI * ds + nnoise *LI * Ns*dt;
|
||||||
|
sinks_tsrc[vecindex] = zero;
|
||||||
|
for (int t = Ntfirst; t < Ntfirst + Ntlocal; t++) {
|
||||||
|
sink_tslice=zero;
|
||||||
|
for (int ivec = 0; ivec < nvec; ivec++) {
|
||||||
|
ExtractSliceLocal(evec3d,eig4d.evec[ivec],0,t,3);
|
||||||
|
sink_tslice += evec3d * perambulator(t, ivec, dk, inoise,dt,ds);
|
||||||
|
}
|
||||||
|
#ifdef USE_LOCAL_SLICES
|
||||||
|
InsertSliceLocal(sink_tslice,sinks_tsrc[vecindex],0,t-Ntfirst,Grid::QCD::Tdir);
|
||||||
|
#else
|
||||||
|
InsertSlice(sink_tslice,sinks_tsrc[vecindex],t,Grid::QCD::Tdir);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@ -68,6 +68,43 @@ std::vector<std::string> Tperambulator_l<FImpl>::getOutput(void)
|
|||||||
template <typename FImpl>
|
template <typename FImpl>
|
||||||
void Tperambulator_l<FImpl>::setup(void)
|
void Tperambulator_l<FImpl>::setup(void)
|
||||||
{
|
{
|
||||||
|
/*
|
||||||
|
std::cout << "Compute perambulator from timeslice " << tsrc << std::endl;
|
||||||
|
|
||||||
|
LatticeSpinColourVector dist_source(grid4d);
|
||||||
|
LatticeSpinColourVector tmp2(grid4d);
|
||||||
|
LatticeSpinColourVector tmp3d(grid3d);
|
||||||
|
LatticeColourVector tmp3d_nospin(grid3d);
|
||||||
|
LatticeColourVector evec3d(grid3d);
|
||||||
|
LatticeColourVector tmp_nospin(grid4d);
|
||||||
|
|
||||||
|
LatticeColourVector result_tmp(grid3d);
|
||||||
|
|
||||||
|
LatticeSpinVector peramb_tmp(grid4d);
|
||||||
|
LatticeFermion result(grid4d); result=zero; //Fermion = SpinColourVector!!!
|
||||||
|
LatticeFermion result_single_component(grid4d); result_single_component=zero; //Fermion = SpinColourVector!!!
|
||||||
|
LatticeColourVector result_nospin(grid4d); result_nospin=zero; //Fermion = SpinColourVector!!!
|
||||||
|
LatticeColourVector result_3d(grid3d); result_3d=zero; //Fermion = SpinColourVector!!!
|
||||||
|
LatticeFermion result_test(grid3d); result_test=zero; //Fermion = SpinColourVector!!!
|
||||||
|
|
||||||
|
|
||||||
|
Real mass=SPar.mass; // TODO Infile
|
||||||
|
Real M5 =SPar.M5; // TODO Infile
|
||||||
|
std::cout << "init RBG " << std::endl;
|
||||||
|
GridRedBlackCartesian RBGrid(grid4d);
|
||||||
|
std::cout << "init RBG done" << std::endl;
|
||||||
|
|
||||||
|
GridCartesian * FGrid = SpaceTimeGrid::makeFiveDimGrid(DPar.Ls,grid4d);
|
||||||
|
GridRedBlackCartesian * FrbGrid = SpaceTimeGrid::makeFiveDimRedBlackGrid(DPar.Ls,grid4d);
|
||||||
|
|
||||||
|
typedef DomainWallFermionR FermionAction;
|
||||||
|
|
||||||
|
FermionAction Dop(Umu,*FGrid,*FrbGrid,*grid4d,RBGrid,mass,M5);
|
||||||
|
|
||||||
|
MdagMLinearOperator<FermionAction,LatticeFermion> HermOp(Dop);
|
||||||
|
ConjugateGradient<LatticeFermion> CG(SPar.CGPrecision,SPar.MaxIterations);
|
||||||
|
SchurRedBlackDiagMooeeSolve<LatticeFermion> SchurSolver(CG);
|
||||||
|
*/
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -75,7 +112,68 @@ void Tperambulator_l<FImpl>::setup(void)
|
|||||||
template <typename FImpl>
|
template <typename FImpl>
|
||||||
void Tperambulator_l<FImpl>::execute(void)
|
void Tperambulator_l<FImpl>::execute(void)
|
||||||
{
|
{
|
||||||
|
/*
|
||||||
|
for (int inoise = 0; inoise < nnoise; inoise++) {
|
||||||
|
for (int dk = 0; dk < LI; dk++) {
|
||||||
|
for (int dt = 0; dt < Nt_inv; dt++) {
|
||||||
|
if(full_tdil) dt=tsrc; //this works for now, as longs as tsrc=0, but will crash otherwise!
|
||||||
|
for (int ds = 0; ds < Ns; ds++) {
|
||||||
|
std::cout << "LapH source vector from noise " << inoise << " and dilution component (d_k,d_t,d_alpha) : (" << dk << ","<< dt << "," << ds << ")" << std::endl;
|
||||||
|
dist_source = zero;
|
||||||
|
tmp3d_nospin = zero;
|
||||||
|
evec3d = zero;
|
||||||
|
for (int it = dt; it < Nt; it += TI){
|
||||||
|
if( it >= Ntfirst && it < Ntfirst + Ntlocal ) {
|
||||||
|
for (int ik = dk; ik < nvec; ik += LI){
|
||||||
|
for (int is = ds; is < Ns; is += Ns){ //at the moment, full spin dilution is enforced
|
||||||
|
std::cout << "LapH source vector from noise " << it << " and dilution component (d_k,d_t,d_alpha) : (" << ik << ","<< is << ")" << std::endl;
|
||||||
|
ExtractSliceLocal(evec3d,eig4d.evec[ik],0,it,3);
|
||||||
|
tmp3d_nospin = evec3d * noises[inoise][it][ik]()(is)(); //noises do not have to be a spin vector
|
||||||
|
tmp3d=zero;
|
||||||
|
pokeSpin(tmp3d,tmp3d_nospin,is);
|
||||||
|
tmp2=zero;
|
||||||
|
#ifdef USE_LOCAL_SLICES
|
||||||
|
InsertSliceLocal(tmp3d,tmp2,0,it-Ntfirst,Grid::QCD::Tdir);
|
||||||
|
#else
|
||||||
|
InsertSlice(tmp3d,tmp2,it,Grid::QCD::Tdir);
|
||||||
|
#endif
|
||||||
|
dist_source += tmp2;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
std::cout << "Inversion for noise " << inoise << " and dilution component (d_k,d_t,d_alpha) : (" << dk << ","<< dt << "," << ds << ")" << std::endl;
|
||||||
|
result=zero;
|
||||||
|
LatticeFermion src5(FGrid);
|
||||||
|
LatticeFermion sol5(FGrid);
|
||||||
|
Dop.ImportPhysicalFermionSource(dist_source,src5);
|
||||||
|
SchurSolver(Dop,src5,sol5);
|
||||||
|
Dop.ExportPhysicalFermionSolution(sol5,result); //These are the meson sinks
|
||||||
|
if (compute_current_sink)
|
||||||
|
current_sink[inoise+nnoise*(dk+LI*(dt+Nt_inv*ds))] = result;
|
||||||
|
std::cout << "Contraction of perambulator from noise " << inoise << " and dilution component (d_k,d_t,d_alpha) : (" << dk << ","<< dt << "," << ds << ")" << std::endl;
|
||||||
|
for (int is = 0; is < Ns; is++) {
|
||||||
|
result_nospin = peekSpin(result,is);
|
||||||
|
for (int t = Ntfirst; t < Ntfirst + Ntlocal; t++) {
|
||||||
|
#ifdef USE_LOCAL_SLICES
|
||||||
|
ExtractSliceLocal(result_3d,result_nospin,0,t-Ntfirst,Grid::QCD::Tdir);
|
||||||
|
#else
|
||||||
|
ExtractSlice(result_3d,result_nospin,t,3);
|
||||||
|
#endif
|
||||||
|
for (int ivec = 0; ivec < nvec; ivec++) {
|
||||||
|
ExtractSliceLocal(evec3d,eig4d.evec[ivec],0,t,3);
|
||||||
|
pokeSpin(perambulator(t, ivec, dk, inoise,dt,ds),innerProduct(evec3d, result_3d),is);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
std::cout << "perambulator done" << std::endl;
|
||||||
|
perambulator.SliceShare( grid3d, grid4d );
|
||||||
|
|
||||||
|
// THIS IS WHERE WE WANT TO SAVE THE PERAMBULATORS TO DISK
|
||||||
|
perambulator.WriteTemporary(std::string(pszPerambPack));
|
||||||
|
*/
|
||||||
}
|
}
|
||||||
|
|
||||||
END_MODULE_NAMESPACE
|
END_MODULE_NAMESPACE
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user