From b302ad3d49ca6f7ef803424f07a746afe180b247 Mon Sep 17 00:00:00 2001
From: Peter Boyle <paboyle@ph.ed.ac.uk>
Date: Thu, 23 Nov 2023 18:20:15 -0500
Subject: [PATCH] multiRHS test in place, passes Yay!

---
 tests/debug/Test_general_coarse.cc | 31 +++++++++++++++++++++++++++++-
 1 file changed, 30 insertions(+), 1 deletion(-)

diff --git a/tests/debug/Test_general_coarse.cc b/tests/debug/Test_general_coarse.cc
index 6f7f0d6d..54097bd9 100644
--- a/tests/debug/Test_general_coarse.cc
+++ b/tests/debug/Test_general_coarse.cc
@@ -229,7 +229,36 @@ int main (int argc, char ** argv)
   std::cout<<GridLogMessage<<"*******************************************"<<std::endl;
   std::cout<<GridLogMessage<<"*******************************************"<<std::endl;
   std::cout<<GridLogMessage<<"*******************************************"<<std::endl;
-  
+
+  //  Create a higher dim coarse grid
+  const int nrhs=vComplex::Nsimd();
+
+  Coordinate mpi=GridDefaultMpi();
+  Coordinate rhMpi ({1,1,mpi[0],mpi[1],mpi[2],mpi[3]});
+  Coordinate rhLatt({nrhs,1,clatt[0],clatt[2],clatt[2],clatt[3]});
+  Coordinate rhSimd({nrhs,1, 1,1,1,1});
+
+  GridCartesian *CoarseMrhs = new GridCartesian(rhLatt,rhSimd,rhMpi); 
+
+  MultiGeneralCoarsenedMatrix mrhs(LittleDiracOp,CoarseMrhs);
+
+  {
+    GridParallelRNG          rh_CRNG(CoarseMrhs);rh_CRNG.SeedFixedIntegers(cseeds);
+    CoarseVector rh_phi(CoarseMrhs);
+    CoarseVector rh_res(CoarseMrhs);
+    random(rh_CRNG,rh_phi);
+    mrhs.M(rh_phi,rh_res);
+
+    for(int r=0;r<nrhs;r++){
+      ExtractSlice(phi,rh_phi,r,0);
+      ExtractSlice(chi,rh_res,r,0);
+      LittleDiracOp.M(phi,Aphi);
+      std::cout << r << " mrhs " << norm2(chi)<<std::endl;
+      std::cout << r << " srhs " << norm2(Aphi)<<std::endl;
+      chi=chi-Aphi;
+      std::cout << r << " diff " << norm2(chi)<<std::endl;
+    }
+  }
   Grid_finalize();
   return 0;
 }