added wrapper for GFlowForward

marcuspetschlies · Sep 3, 2024 · a38f0f9 · a38f0f9
1 parent 907ce3e
commit a38f0f9
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/quda_interface.c b/quda_interface.c
@@ -3006,14 +3006,40 @@ void _performGFlowAdjoint ( double * const h_out, double * const h_in, QudaGauge
     memcpy ( tempSpinor, h_out, VOLUME*24*sizeof(double) );
   }
 
-  performGFlowAdjoint ( (void *)h_out, (void *)h_in, &inv_param, smear_param, mb, nb, store );
+  performGFlowAdjoint ( (void *)h_out, (void *)tempSpinor, &inv_param, smear_param, mb, nb, store );
 
   if ( h_out != NULL && h_in != NULL )
   {
     reorder_spinor_fromQuda ( h_out, inv_param.cpu_prec, 0 );
   }
 }  /* end of _performGFlowAdjoint */
 
+/**********************************************************************/
+/**********************************************************************/
+
+/**********************************************************************
+ * wrapper for GFlowForward
+ **********************************************************************/
+void _performGFlowForward ( double * const h_out, double * const h_in, QudaGaugeSmearParam *smear_param, int const update_gauge )
+{
+  if ( h_out != NULL && h_in != NULL )
+  {
+    if ( h_out != h_in )
+    {
+      memcpy ( h_out, h_in, VOLUME*24*sizeof(double) );
+    }
+    reorder_spinor_toQuda ( h_out, inv_param.cpu_prec, 0 );
+
+    memcpy ( tempSpinor, h_out, VOLUME*24*sizeof(double) );
+  }
+
+  performGFlowForward ( (void *)h_out, (void *)h_in, &inv_param, smear_param, update_gauge );
+
+  if ( h_out != NULL && h_in != NULL )
+  {
+    reorder_spinor_fromQuda ( h_out, inv_param.cpu_prec, 0 );
+  }
+}  /* end of _performGFlowForward */
 
 
 
diff --git a/quda_interface.h b/quda_interface.h
@@ -194,4 +194,6 @@ void _performWuppertalnStep ( double * const h_out, double * const h_in, unsigne
 
 void _performGFlowAdjoint ( double * const h_out, double * const h_in, QudaGaugeSmearParam *smear_param, int const mb, int const nb, int const store );
 
+void _performGFlowForward ( double * const h_out, double * const h_in, QudaGaugeSmearParam *smear_param, int const update_gauge );
+
 #endif /* QUDA_INTERFACE_H_ */
Original file line number	Diff line number	Diff line change
Expand Up		@@ -194,4 +194,6 @@ void _performWuppertalnStep ( double * const h_out, double * const h_in, unsigne

		void _performGFlowAdjoint ( double * const h_out, double * const h_in, QudaGaugeSmearParam *smear_param, int const mb, int const nb, int const store );

		void _performGFlowForward ( double * const h_out, double * const h_in, QudaGaugeSmearParam *smear_param, int const update_gauge );

		#endif /* QUDA_INTERFACE_H_ */