OMP volume loop

etmc · Dec 8, 2023 · 0df1801 · 0df1801
1 parent 020dfa3
commit 0df1801
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/monomial/cloverdet_monomial.c b/monomial/cloverdet_monomial.c
@@ -109,16 +109,19 @@ void cloverdet_derivative(const int id, hamiltonian_field_t * const hf) {
     if(!mnl->even_odd_flag) {
       fatal_error("QUDA support only even_odd_flag",__func__);
     }
-  #ifdef TM_USE_QUDA
+#ifdef TM_USE_QUDA
     if (g_debug_level > 3) {
 #ifdef TM_USE_MPI
       // FIXME: here we do not need to set to zero the interior but only the halo
+#ifdef TM_USE_OMP
+      # pragma omp parallel for
+#endif
       for(int i = 0; i < (VOLUMEPLUSRAND + g_dbw2rand);i++) { 
         for(int mu=0;mu<4;mu++) { 
           _zero_su3adj(debug_derivative[i][mu]);
         }
       }
-#endif
+#endif // end setting to zero the halo when using MPI
       // we copy only the interior
       memcpy(debug_derivative[0], hf->derivative[0], 4*VOLUME*sizeof(su3adj));
     }

diff --git a/monomial/cloverdetratio_monomial.c b/monomial/cloverdetratio_monomial.c
@@ -219,10 +219,13 @@ void cloverdetratio_derivative(const int no, hamiltonian_field_t * const hf) {
     if(!mnl->even_odd_flag) {
       fatal_error("QUDA support only even_odd_flag",__func__);
     }
-  #ifdef TM_USE_QUDA
+#ifdef TM_USE_QUDA
     if (g_debug_level > 3) {
 #ifdef TM_USE_MPI
       // FIXME: here we do not need to set to zero the interior but only the halo
+#ifdef TM_USE_OMP
+      # pragma omp parallel for
+#endif // end setting to zero the halo when using MPI
       for(int i = 0; i < (VOLUMEPLUSRAND + g_dbw2rand);i++) { 
         for(int mu=0;mu<4;mu++) { 
           _zero_su3adj(debug_derivative[i][mu]);