Merge branch 'feature/tm_force' of github.com:lattice/quda into featu…

…re/tm_force
lattice · Dec 7, 2023 · 3aed5e0 · 3aed5e0
2 parents d80a476 + c1ee1f7
commit 3aed5e0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/lib/clover_outer_product.cu b/lib/clover_outer_product.cu
@@ -43,12 +43,12 @@ namespace quda {
       apply(device::get_default_stream());
 
       for (int i=3; i>=0; i--) {
+        dir = i;
         if (!commDimPartitioned(i)) continue;
         strcpy(aux, aux2);
         strcat(aux, ",exterior,dir=");
         strcat(aux, dir == 0 ? "0" : dir == 1 ? "1" : dir == 2 ? "2" : "3");
         kernel = EXTERIOR;
-        dir = i;
         apply(device::get_default_stream());
       }
     }

diff --git a/tests/TMCloverForce_test.cpp b/tests/TMCloverForce_test.cpp
@@ -152,6 +152,7 @@ void TMCloverForce_test()
   quda::GaugeField Mom_qdp(param);
 
   // initialize some data in cpuMom
+  // we need to set the mom to zero because computeTMCloverForceQuda is overwriting the momentum
   createMomCPU(Mom_ref_milc.data(), gauge_param.cpu_prec, 0.0);
   if (gauge_order == QUDA_MILC_GAUGE_ORDER) Mom_milc.copy(Mom_ref_milc);
   if (gauge_order == QUDA_QDP_GAUGE_ORDER) Mom_qdp.copy(Mom_ref_milc);
@@ -219,7 +220,7 @@ void TMCloverForce_test()
   if (getTuning() == QUDA_TUNE_YES)
     computeTMCloverForceQuda(mom, in[0].data(), in0[0].data(), coeff, nvector,  &gauge_param, &inv_param, detratio);
 
-  printf("Device function computed\n");
+  printfQuda("Device function computed\n");
   quda::host_timer_t host_timer;
   double time_sec = 0.0;
   // Multiple execution to exclude warmup time in the first run