Added ImportUnphysicalFermion and included appropriate logic for 5d w vectors in A2A code

extras/Hadrons/AllToAllVectors.hpp

@@ -51,14 +51,14 @@ class A2AModesSchurDiagTwo
         v_high_4d.resize(size_4d, grid_4d);
     }
 
-    void high_modes(Field &source_5d, Field &source_4d, int i)
+    void high_modes(Field &source_5d, Field &w_source_5d, Field &source_4d, int i)
     {
         int i5d;
         LOG(Message) << "A2A high modes for i = " << i << std::endl;
         i5d = 0;
         if (return_5d) i5d = i;
         this->high_mode_v(action, solver, source_5d, v_high_5d[i5d], v_high_4d[i]);
-        this->high_mode_w(source_5d, source_4d, w_high_5d[i5d], w_high_4d[i]);
+        this->high_mode_w(w_source_5d, source_4d, w_high_5d[i5d], w_high_4d[i]);
     }
 
     void return_v(int i, Field &vout_5d, Field &vout_4d)
@@ -181,9 +181,9 @@ class A2AModesSchurDiagTwo
         action.ExportPhysicalFermionSolution(vout_5d, vout_4d);
     }
 
-    void high_mode_w(const Field &source_5d, const Field &source_4d, Field &wout_5d, Field &wout_4d)
+    void high_mode_w(const Field &w_source_5d, const Field &source_4d, Field &wout_5d, Field &wout_4d)
     {
-        wout_5d = source_5d;
+        wout_5d = w_source_5d;
         wout_4d = source_4d;
     }
 };

extras/Hadrons/Modules/MSolver/A2AVectors.hpp

@@ -130,6 +130,7 @@ void TA2AVectors<FImpl, nBasis>::setup(void)
     auto &action = envGet(FMat, par().action);
 
     envTmpLat(FermionField, "ferm_src", Ls_);
+    envTmpLat(FermionField, "unphys_ferm", Ls_);
     envTmpLat(FermionField, "tmp");
     envTmpLat(FermionField, "tmp2");
 
@@ -183,6 +184,7 @@ void TA2AVectors<FImpl, nBasis>::execute(void)
     int Nsrc = par().sources.size();
 
     envGetTmp(FermionField, ferm_src);
+    envGetTmp(FermionField, unphys_ferm);
     envGetTmp(FermionField, tmp);
     envGetTmp(FermionField, tmp2);
 
@@ -205,6 +207,7 @@ void TA2AVectors<FImpl, nBasis>::execute(void)
                 {
                     PropToFerm<FImpl>(tmp, prop_src, s, c);
                     action.ImportPhysicalFermionSource(tmp, ferm_src);
+                    action.ImportUnphysicalFermion(tmp, unphys_ferm);
                 }
             }
             // source conversion for 5D sources
@@ -218,10 +221,11 @@ void TA2AVectors<FImpl, nBasis>::execute(void)
                 {
                     PropToFerm<FImpl>(ferm_src, prop_src, s, c);
                     action.ExportPhysicalFermionSolution(ferm_src, tmp);
+                    unphys_ferm = ferm_src;
                 }
             }
             LOG(Message) << "a2areturn.high_modes Ncount = " << N_count << std::endl;
-            a2areturn.high_modes(ferm_src, tmp, N_count);
+            a2areturn.high_modes(ferm_src, unphys_ferm, tmp, N_count);
             N_count++;
         }
 }

lib/qcd/action/fermion/CayleyFermion5D.cc

@@ -79,6 +79,21 @@ void CayleyFermion5D<Impl>::ExportPhysicalFermionSource(const FermionField &solu
   axpby_ssp_pminus(tmp, 1., tmp       , 1., solution5d, 0, Ls-1);
   ExtractSlice(exported4d, tmp, 0, 0);
 }
+template<class Impl>
+void CayleyFermion5D<Impl>::ImportUnphysicalFermion(const FermionField &input4d,FermionField &imported5d)
+{
+  int Ls = this->Ls;
+  FermionField tmp(this->FermionGrid());
+  conformable(imported5d._grid,this->FermionGrid());
+  conformable(input4d._grid   ,this->GaugeGrid());
+  tmp = zero;
+  InsertSlice(input4d, tmp, 0   , 0);
+  InsertSlice(input4d, tmp, Ls-1, 0);
+  axpby_ssp_pplus (tmp, 0., tmp, 1., tmp, 0, 0);
+  axpby_ssp_pminus(tmp, 0., tmp, 1., tmp, Ls-1, Ls-1);
+  imported5d=tmp;
+}
+
 template<class Impl>  
 void CayleyFermion5D<Impl>::ImportPhysicalFermionSource(const FermionField &input4d,FermionField &imported5d)
 {

lib/qcd/action/fermion/CayleyFermion5D.h

@@ -91,6 +91,7 @@ namespace Grid {
       virtual void ExportPhysicalFermionSolution(const FermionField &solution5d,FermionField &exported4d);
       virtual void ExportPhysicalFermionSource(const FermionField &solution5d, FermionField &exported4d);
       virtual void ImportPhysicalFermionSource(const FermionField &input4d,FermionField &imported5d);
+      virtual void ImportUnphysicalFermion(const FermionField &solution5d, FermionField &exported4d);
 
       /////////////////////////////////////////////////////
       // Instantiate different versions depending on Impl

lib/qcd/action/fermion/FermionOperator.h

@@ -162,13 +162,17 @@ namespace Grid {
       {
 	imported = input;
       };
+      virtual void ImportUnphysicalFermion(const FermionField &input,FermionField &imported)
+      {
+	imported=input;
+      };
       virtual void ExportPhysicalFermionSolution(const FermionField &solution,FermionField &exported)
       {
 	exported=solution;
       };
-      virtual void ExportPhysicalFermionSource(const FermionField &solution, FermionField &exported)
+      virtual void ExportPhysicalFermionSource(const FermionField &solution,FermionField &exported)
       {
-	exported = solution;
+	exported=solution;
       };
     };