diff --git a/Grid/qcd/utils/A2Autils.h b/Grid/qcd/utils/A2Autils.h index 81c53800..73b9d16f 100644 --- a/Grid/qcd/utils/A2Autils.h +++ b/Grid/qcd/utils/A2Autils.h @@ -43,9 +43,9 @@ public: static void NucleonFieldMom(Eigen::Tensor &mat, - const FermionField *one, - const FermionField *two, - const FermionField *three, + const std::vector &one, + const std::vector &two, + const std::vector &three, const std::vector &mom, int parity, int orthogdim); @@ -112,9 +112,9 @@ public: template void A2Autils::NucleonFieldMom(Eigen::Tensor &mat, - const FermionField *one, - const FermionField *two, - const FermionField *three, + const std::vector &one, + const std::vector &two, + const std::vector &three, const std::vector &mom, int parity, int orthogdim) diff --git a/Hadrons/Modules/MDistil/BC2.hpp b/Hadrons/Modules/MDistil/BC2.hpp index 5c3433cc..d3c4475a 100644 --- a/Hadrons/Modules/MDistil/BC2.hpp +++ b/Hadrons/Modules/MDistil/BC2.hpp @@ -56,7 +56,7 @@ BEGIN_MODULE_NAMESPACE(MDistil) // 3 - s - free spin index // 4 - i - left distillation mode index // 5 - j - middle distillation mode index - // 6 - k - left distillation mode index + // 6 - k - right distillation mode index // template // using BaryonTensorSet = Eigen::TensorMap>; @@ -133,6 +133,7 @@ std::vector TBC2::getOutput(void) template void TBC2::setup(void) { + if(!mom_.size()) { for (auto &pstr: par().mom) { auto p = strToVec(pstr); @@ -143,8 +144,19 @@ void TBC2::setup(void) } mom_.push_back(p); } - envCache(std::vector, momphName_, 1, - par().mom.size(), envGetGrid(ComplexField)); + } + //envCache(std::vector, momphName_, 1, par().mom.size(), envGetGrid(ComplexField)); + static GridCartesian * MyGrid{env().getGrid()}; + if( MyGrid == envGetGrid(ComplexField) ) + LOG(Message) << "envGetGrid(ComplexField) == env().getGrid()" << std::endl; + else + LOG(Message) << "envGetGrid(ComplexField) != env().getGrid()" << std::endl; + envTmp(std::vector, "ph", 1, std::vector()); + envGetTmp(std::vector, ph); + if(!ph.size()) { + for (unsigned int j = 0; j < par().mom.size(); ++j) + ph.push_back(ComplexField(MyGrid)); + } envTmpLat(ComplexField, "coor"); } @@ -153,16 +165,14 @@ void TBC2::setup(void) template void TBC2::execute(void) { - - auto &one = envGet(std::vector, par().one); auto &two = envGet(std::vector, par().two); auto &three = envGet(std::vector, par().three); const std::string &output{par().output}; - int N_1 = one.size(); - int N_2 = two.size(); - int N_3 = three.size(); + int N_1 = static_cast(one.size()); + int N_2 = static_cast(two.size()); + int N_3 = static_cast(three.size()); LOG(Message) << "Computing distillation baryon fields" << std::endl; LOG(Message) << "One: '" << par().one << "' Two: '" << par().two << "' Three: '" << par().three << "'" << std::endl; @@ -173,13 +183,14 @@ void TBC2::execute(void) } - int Nmom = mom_.size(); + int Nmom = static_cast(mom_.size()); const int Nt{env().getDim(Tdir)}; int parity = 1; int orthogDim=3; - auto &ph = envGet(std::vector, momphName_); + //auto &ph = envGet(std::vector, momphName_); + envGetTmp(std::vector, ph); if (!hasPhase_) { @@ -201,11 +212,11 @@ void TBC2::execute(void) hasPhase_ = true; stopTimer("Momentum phases"); } - envCache(std::vector, momphName_, 1, mom_.size(), envGetGrid(ComplexField)); + //envCache(std::vector, momphName_, 1, mom_.size(), envGetGrid(ComplexField)); Eigen::Tensor m(Nmom,Nt,N_1,N_2,N_3,4); - A2Autils::NucleonFieldMom(m, &one[0], &two[0], &three[0], ph, parity, orthogDim); - //A2Autils::NucleonFieldMom(m, one, two, three, ph, parity, orthogDim); + //A2Autils::NucleonFieldMom(m, &one[0], &two[0], &three[0], ph, parity, orthogDim); + A2Autils::NucleonFieldMom(m, one, two, three, ph, parity, orthogDim); for (int is=0 ; is < 4 ; is++){ for (int t=0 ; t < Nt ; t++){ std::cout << "BaryonField(is=" << is << ",t=" << t << ") = " << m(0,t,0,0,0,is) << std::endl;