#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <quda.h>
#include <quda_milc_interface.h>
#include <quda_internal.h>
#include <color_spinor_field.h>
#include <string.h>
#include <unitarization_links.h>
#include <ks_improved_force.h>
#include <dslash_quda.h>
#include <vector>
#include <fstream>

Classes
struct	mgInputStruct

struct	milcMultigridPack

Macros
#define	MAX(a, b) ((a) > (b) ? (a) : (b))

#define	PUSH_RANGE(name, cid)

#define	POP_RANGE

#define	QUDAMILC_VERBOSE 1

Functions
template<bool start>
void	qudamilc_called (const char *func, QudaVerbosity verb)

template<bool start>
void	qudamilc_called (const char *func)

void	qudaSetMPICommHandle (void *mycomm)

void	qudaInit (QudaInitArgs_t input)

void	qudaFinalize ()

void	qudaSetLayout (QudaLayout_t input)

void *	qudaAllocatePinned (size_t bytes)

void	qudaFreePinned (void *ptr)

void *	qudaAllocateManaged (size_t bytes)

void	qudaFreeManaged (void *ptr)

void	qudaHisqParamsInit (QudaHisqParams_t params)

void	qudaLoadKSLink (int prec, QudaFatLinkArgs_t fatlink_args, const double act_path_coeff[6], void inlink, void fatlink, void *longlink)

void	qudaLoadUnitarizedLink (int prec, QudaFatLinkArgs_t fatlink_args, const double act_path_coeff[6], void inlink, void fatlink, void *ulink)

void	qudaHisqForce (int prec, int num_terms, int num_naik_terms, double dt, double coeff, void quark_field, const double level2_coeff[6], const double fat7_coeff[6], const void const w_link, const void const v_link, const void const u_link, void const milc_momentum)

void	qudaAsqtadForce (int prec, const double act_path_coeff[6], const void const one_link_src[4], const void const naik_src[4], const void const link, void const milc_momentum)

void	qudaComputeOprod (int prec, int num_terms, int num_naik_terms, double coeff, double scale, void quark_field, void *oprod[3])

void	qudaUpdateUPhasedPipeline (int prec, double eps, QudaMILCSiteArg_t *arg, int phase_in, int want_gaugepipe)

void	qudaUpdateUPhased (int prec, double eps, QudaMILCSiteArg_t *arg, int phase_in)

void	qudaUpdateU (int prec, double eps, QudaMILCSiteArg_t *arg)

void	qudaRephase (int prec, void *gauge, int flag, double i_mu)

void	qudaUnitarizeSU3Phased (int prec, double tol, QudaMILCSiteArg_t *arg, int phase_in)

void	qudaUnitarizeSU3 (int prec, double tol, QudaMILCSiteArg_t *arg)

void	qudaMomLoad (int prec, QudaMILCSiteArg_t *arg)

void	qudaMomSave (int prec, QudaMILCSiteArg_t *arg)

double	qudaMomAction (int prec, QudaMILCSiteArg_t *arg)

void	qudaGaugeForcePhased (int precision, int num_loop_types, double milc_loop_coeff[3], double eb3, QudaMILCSiteArg_t *arg, int phase_in)

void	qudaGaugeForce (int precision, int num_loop_types, double milc_loop_coeff[3], double eb3, QudaMILCSiteArg_t *arg)

void	setDeflationParam (QudaPrecision ritz_prec, QudaFieldLocation location_ritz, QudaMemoryType mem_type_ritz, QudaExtLibType deflation_ext_lib, char vec_infile[], char vec_outfile[], QudaEigParam *df_param)

void	qudaMultishiftInvert (int external_precision, int quda_precision, int num_offsets, double const offset, QudaInvertArgs_t inv_args, const double target_residual[], const double target_fermilab_residual[], const void const fatlink, const void const longlink, void source, void *solutionArray, double const final_residual, double const final_fermilab_residual, int num_iters)

void	qudaInvert (int external_precision, int quda_precision, double mass, QudaInvertArgs_t inv_args, double target_residual, double target_fermilab_residual, const void const fatlink, const void const longlink, void source, void solution, double const final_residual, double const final_fermilab_residual, int *num_iters)

void	qudaDslash (int external_precision, int quda_precision, QudaInvertArgs_t inv_args, const void const fatlink, const void const longlink, void src, void dst, int *num_iters)

void	qudaInvertMsrc (int external_precision, int quda_precision, double mass, QudaInvertArgs_t inv_args, double target_residual, double target_fermilab_residual, const void const fatlink, const void const longlink, void sourceArray, void solutionArray, double const final_residual, double const final_fermilab_residual, int *num_iters, int num_src)

void	qudaEigCGInvert (int external_precision, int quda_precision, double mass, QudaInvertArgs_t inv_args, double target_residual, double target_fermilab_residual, const void const fatlink, const void const longlink, void source, void solution, QudaEigArgs_t eig_args, const int rhs_idx, const int last_rhs_flag, double const final_residual, double const final_fermilab_residual, int *num_iters)

void	milcSetMultigridEigParam (QudaEigParam &mg_eig_param, mgInputStruct &input_struct, int level)

void	milcSetMultigridParam (milcMultigridPack mg_pack, QudaPrecision host_precision, QudaPrecision device_precision, QudaPrecision device_precision_sloppy, double mass, const char const mg_param_file)

void *	qudaMultigridCreate (int external_precision, int quda_precision, double mass, QudaInvertArgs_t inv_args, const void const fatlink, const void const longlink, const char *const mg_param_file)

void	qudaInvertMG (int external_precision, int quda_precision, double mass, QudaInvertArgs_t inv_args, double target_residual, double target_fermilab_residual, const void const fatlink, const void const longlink, void mg_pack_ptr, int mg_rebuild_type, void source, void solution, double const final_residual, double const final_fermilab_residual, int num_iters)

void	qudaMultigridDestroy (void *mg_pack_ptr)

void *	qudaCreateGaugeField (void *gauge, int geometry, int precision)

void	qudaSaveGaugeField (void gauge, void inGauge)

void	qudaDestroyGaugeField (void *gauge)

void	setInvertParam (QudaInvertParam &invertParam, QudaInvertArgs_t &inv_args, int external_precision, int quda_precision, double kappa, double reliable_delta)

void	qudaCloverForce (void mom, double dt, void x, void p, double coeff, double kappa, double ck, int nvec, double multiplicity, void *gauge, int precision, QudaInvertArgs_t inv_args)

void	setGaugeParams (QudaGaugeParam &qudaGaugeParam, const int dim[4], QudaInvertArgs_t &inv_args, int external_precision, int quda_precision)

void	qudaLoadGaugeField (int external_precision, int quda_precision, QudaInvertArgs_t inv_args, const void *milc_link)

void	qudaFreeGaugeField ()

void	qudaLoadCloverField (int external_precision, int quda_precision, QudaInvertArgs_t inv_args, void milc_clover, void milc_clover_inv, QudaSolutionType solution_type, QudaSolveType solve_type, QudaInverterType inverter, double clover_coeff, int compute_trlog, double *trlog)

void	qudaFreeCloverField ()

void	qudaCloverInvert (int external_precision, int quda_precision, double kappa, double clover_coeff, QudaInvertArgs_t inv_args, double target_residual, double target_fermilab_residual, const void link, void clover, void cloverInverse, void source, void solution, double const final_residual, double const final_fermilab_residual, int num_iters)

void	qudaEigCGCloverInvert (int external_precision, int quda_precision, double kappa, double clover_coeff, QudaInvertArgs_t inv_args, double target_residual, double target_fermilab_residual, const void link, void clover, void cloverInverse, void source, void solution, QudaEigArgs_t eig_args, const int rhs_idx, const int last_rhs_flag, double const final_residual, double const final_fermilab_residual, int num_iters)

void	qudaCloverMultishiftInvert (int external_precision, int quda_precision, int num_offsets, double const offset, double kappa, double clover_coeff, QudaInvertArgs_t inv_args, const double target_residual_offset, const void milc_link, void milc_clover, void milc_clover_inv, void source, void *solutionArray, double const final_residual, int *num_iters)

void	qudaGaugeFixingOVR (int precision, unsigned int gauge_dir, int Nsteps, int verbose_interval, double relax_boost, double tolerance, unsigned int reunit_interval, unsigned int stopWtheta, void *milc_sitelink)
	Gauge fixing with overrelaxation with support for single and multi GPU. More...

void	qudaGaugeFixingFFT (int precision, unsigned int gauge_dir, int Nsteps, int verbose_interval, double alpha, unsigned int autotune, double tolerance, unsigned int stopWtheta, void *milc_sitelink)
	Gauge fixing with Steepest descent method with FFTs with support for single GPU only. More...

Macro Definition Documentation

◆ MAX

#define MAX	(	a,
		b
	)	((a) > (b) ? (a) : (b))

Definition at line 16 of file milc_interface.cpp.

◆ POP_RANGE

#define POP_RANGE

Definition at line 47 of file milc_interface.cpp.

◆ PUSH_RANGE

#define PUSH_RANGE	(	name,
		cid
	)

Definition at line 46 of file milc_interface.cpp.

◆ QUDAMILC_VERBOSE

#define QUDAMILC_VERBOSE 1

Definition at line 70 of file milc_interface.cpp.

Function Documentation

◆ milcSetMultigridEigParam()

void milcSetMultigridEigParam	(	QudaEigParam &	mg_eig_param,
		mgInputStruct &	input_struct,
		int	level
	)

Definition at line 1718 of file milc_interface.cpp.

◆ milcSetMultigridParam()

void milcSetMultigridParam	(	milcMultigridPack *	mg_pack,
		QudaPrecision	host_precision,
		QudaPrecision	device_precision,
		QudaPrecision	device_precision_sloppy,
		double	mass,
		const char *const	mg_param_file
	)

Definition at line 1758 of file milc_interface.cpp.

◆ qudaAllocateManaged()

void* qudaAllocateManaged ( size_t bytes )

Allocate managed memory to reduce CPU-GPU transfers

Parameters

bytes The size of the requested allocation

Returns: Pointer to allocated memory

Definition at line 165 of file milc_interface.cpp.

◆ qudaAllocatePinned()

void* qudaAllocatePinned ( size_t bytes )

Allocate pinned memory suitable for CPU-GPU transfers

Parameters

bytes The size of the requested allocation

Returns: Pointer to allocated memory

Definition at line 161 of file milc_interface.cpp.

◆ qudaAsqtadForce()

void qudaAsqtadForce	(	int	precision,
		const double	act_path_coeff[6],
		const void *const	one_link_src[4],
		const void *const	naik_src[4],
		const void *const	link,
		void *const	milc_momentum
	)

Note this interface function has been removed. This stub remains for compatibility only.

Definition at line 313 of file milc_interface.cpp.

◆ qudaCloverForce()

void qudaCloverForce	(	void *	mom,
		double	dt,
		void **	x,
		void **	p,
		double *	coeff,
		double	kappa,
		double	ck,
		int	nvec,
		double	multiplicity,
		void *	gauge,
		int	precision,
		QudaInvertArgs_t	inv_args
	)

Compute the clover force contributions in each dimension mu given the array solution fields, and compute the resulting momentum field.

Parameters

mom	Momentum matrix
dt	Integrating step size
x	Array of solution vectors
p	Array of intermediate vectors
coeff	Array of residues for each contribution
kappa	kappa parameter
ck	-clover_coefficient * kappa / 8
nvec	Number of vectors
multiplicity	Number of fermions represented by this bilinear
gauge	Gauge Field
precision	Precision of the fields
inv_args	Struct setting some solver metadata

Definition at line 2303 of file milc_interface.cpp.

◆ qudaCloverInvert()

void qudaCloverInvert	(	int	external_precision,
		int	quda_precision,
		double	kappa,
		double	clover_coeff,
		QudaInvertArgs_t	inv_args,
		double	target_residual,
		double	target_fermilab_residual,
		const void *	milc_link,
		void *	milc_clover,
		void *	milc_clover_inv,
		void *	source,
		void *	solution,
		double *const	final_residual,
		double *const	final_fermilab_residual,
		int *	num_iters
	)

Solve Ax=b using a Wilson-Clover operator. All fields are fields passed and returned are host (CPU) field in MILC order. This function creates the gauge and clover field from the host fields. Reliable updates are used with a reliable_delta parameter of 0.1.

Parameters

external_precision	Precision of host fields passed to QUDA (2 - double, 1 - single)
quda_precision	Precision for QUDA to use (2 - double, 1 - single)
kappa	Kappa value
clover_coeff	Clover coefficient
inv_args	Struct setting some solver metadata
target_residual	Target residual
milc_link	Gauge field on the host
milc_clover	Clover field on the host
milc_clover_inv	Inverse clover on the host
clover_coeff	Clover coefficient
source	Right-hand side source field
solution	Solution spinor field
final_residual	True residual returned by the solver
final_residual	True Fermilab residual returned by the solver
num_iters	Number of iterations taken

Definition at line 2492 of file milc_interface.cpp.

◆ qudaCloverMultishiftInvert()

void qudaCloverMultishiftInvert	(	int	external_precision,
		int	quda_precision,
		int	num_offsets,
		double *const	offset,
		double	kappa,
		double	clover_coeff,
		QudaInvertArgs_t	inv_args,
		const double *	target_residual,
		const void *	milc_link,
		void *	milc_clover,
		void *	milc_clover_inv,
		void *	source,
		void **	solutionArray,
		double *const	final_residual,
		int *	num_iters
	)

Solve for multiple shifts (e.g., masses) using a Wilson-Clover operator with multi-shift CG. All fields are fields passed and returned are host (CPU) field in MILC order. This function requires that persistent gauge and clover fields have been created prior. When a pure double-precision solver is requested no reliable updates are used, else reliable updates are used with a reliable_delta parameter of 0.1.

Parameters

external_precision	Precision of host fields passed to QUDA (2 - double, 1 - single)
quda_precision	Precision for QUDA to use (2 - double, 1 - single)
num_offsets	Number of shifts to solve for
offset	Array of shift offset values
kappa	Kappa value
clover_coeff	Clover coefficient
inv_args	Struct setting some solver metadata
target_residual	Array of target residuals per shift
milc_link	Ignored
milc_clover	Ignored
milc_clover_inv	Ignored
clover_coeff	Clover coefficient
source	Right-hand side source field
solutionArray	Array of solution spinor fields
final_residual	Array of true residuals
num_iters	Number of iterations taken

Definition at line 2628 of file milc_interface.cpp.

◆ qudaComputeOprod()

void qudaComputeOprod	(	int	precision,
		int	num_terms,
		int	num_naik_terms,
		double **	coeff,
		double	scale,
		void **	quark_field,
		void *	oprod[3]
	)

Note this interface function has been removed. This stub remains for compatibility only.

Definition at line 322 of file milc_interface.cpp.

◆ qudaCreateGaugeField()

void* qudaCreateGaugeField	(	void *	gauge,
		int	geometry,
		int	precision
	)

Allocate a gauge (matrix) field on the device and optionally download a host gauge field.

Parameters

gauge	The host gauge field (optional - if set to 0 then the gauge field zeroed)
geometry	The geometry of the matrix field to create (1 - scaler, 4 - vector, 6 - tensor)
precision	The precision of the field to be created (2 - double, 1 - single)

Returns: Pointer to the gauge field (cast as a void*)

Definition at line 2271 of file milc_interface.cpp.

◆ qudaDestroyGaugeField()

void qudaDestroyGaugeField ( void * gauge )

Reinterpret gauge as a pointer to cudaGaugeField and call destructor.

Parameters

gauge Gauge field to be freed

Definition at line 2292 of file milc_interface.cpp.

◆ qudaDslash()

void qudaDslash	(	int	external_precision,
		int	quda_precision,
		QudaInvertArgs_t	inv_args,
		const void *const	milc_fatlink,
		const void *const	milc_longlink,
		void *	source,
		void *	solution,
		int *	num_iters
	)

Apply the improved staggered operator to a field. All fields passed and returned are host (CPU) field in MILC order.

Parameters

external_precision	Precision of host fields passed to QUDA (2 - double, 1 - single)
quda_precision	Precision for QUDA to use (2 - double, 1 - single)
inv_args	Struct setting some solver metadata
milc_fatlink	Fat-link field on the host
milc_longlink	Long-link field on the host
source	Right-hand side source field
solution	Solution spinor field

Definition at line 1066 of file milc_interface.cpp.

◆ qudaEigCGCloverInvert()

void qudaEigCGCloverInvert	(	int	external_precision,
		int	quda_precision,
		double	kappa,
		double	clover_coeff,
		QudaInvertArgs_t	inv_args,
		double	target_residual,
		double	target_fermilab_residual,
		const void *	milc_link,
		void *	milc_clover,
		void *	milc_clover_inv,
		void *	source,
		void *	solution,
		QudaEigArgs_t	eig_args,
		const int	rhs_idx,
		const int	last_rhs_flag,
		double *const	final_residual,
		double *const	final_fermilab_residual,
		int *	num_iters
	)

Solve for a system with many RHS using using a Wilson-Clover operator. The solving procedure consists of two computation phases : 1) incremental pahse : call eigCG solver to accumulate low eigenmodes 2) deflation phase : use computed eigenmodes to deflate a regular CG All fields are fields passed and returned are host (CPU) field in MILC order. This function requires that persistent gauge and clover fields have been created prior.

Parameters

external_precision	Precision of host fields passed to QUDA (2 - double, 1 - single)
quda_precision	Precision for QUDA to use (2 - double, 1 - single)
kappa	Kappa value
clover_coeff	Clover coefficient
inv_args	Struct setting some solver metadata
target_residual	Target residual
milc_link	Gauge field on the host
milc_clover	Clover field on the host
milc_clover_inv	Inverse clover on the host
clover_coeff	Clover coefficient
source	Right-hand side source field
solution	Solution spinor field
eig_args	contains info about deflation space
rhs_idx	bookkeep current rhs
last_rhs_flag	is this the last rhs to solve?
final_residual	Array of true residuals
final_relative_residual	Array of true Fermilab residuals
num_iters	Number of iterations taken

Definition at line 2548 of file milc_interface.cpp.

◆ qudaEigCGInvert()

void qudaEigCGInvert	(	int	external_precision,
		int	quda_precision,
		double	mass,
		QudaInvertArgs_t	inv_args,
		double	target_residual,
		double	target_fermilab_residual,
		const void *const	fatlink,
		const void *const	longlink,
		void *	source,
		void *	solution,
		QudaEigArgs_t	eig_args,
		const int	rhs_idx,
		const int	last_rhs_flag,
		double *const	final_residual,
		double *const	final_fermilab_residual,
		int *	num_iters
	)

Solve for a system with many RHS using an improved staggered operator. The solving procedure consists of two computation phases : 1) incremental pahse : call eigCG solver to accumulate low eigenmodes 2) deflation phase : use computed eigenmodes to deflate a regular CG All fields are fields passed and returned are host (CPU) field in MILC order. This function requires that persistent gauge and clover fields have been created prior.

Parameters

external_precision	Precision of host fields passed to QUDA (2 - double, 1 - single)
precision	Precision for QUDA to use (2 - double, 1 - single)
num_offsets	Number of shifts to solve for
offset	Array of shift offset values
inv_args	Struct setting some solver metadata
target_residual	Array of target residuals per shift
target_relative_residual	Array of target Fermilab residuals per shift
milc_fatlink	Fat-link field on the host
milc_longlink	Long-link field on the host
source	Right-hand side source field
solution	Array of solution spinor fields
eig_args	contains info about deflation space
rhs_idx	bookkeep current rhs
last_rhs_flag	is this the last rhs to solve?
final_residual	Array of true residuals
final_relative_residual	Array of true Fermilab residuals
num_iters	Number of iterations taken

Definition at line 1188 of file milc_interface.cpp.

◆ qudaFinalize()

void qudaFinalize ( )

Destroy the QUDA context.

Definition at line 106 of file milc_interface.cpp.

◆ qudaFreeCloverField()

void qudaFreeCloverField ( )

Free the clover field allocated in QUDA.

Definition at line 2480 of file milc_interface.cpp.

◆ qudaFreeGaugeField()

void qudaFreeGaugeField ( )

Free the gauge field allocated in QUDA.

Definition at line 2435 of file milc_interface.cpp.

◆ qudaFreeManaged()

void qudaFreeManaged ( void * ptr )

Free managed memory

Parameters

ptr	Pointer to memory to be free

Definition at line 167 of file milc_interface.cpp.

◆ qudaFreePinned()

void qudaFreePinned ( void * ptr )

Free pinned memory

Parameters

ptr	Pointer to memory to be free

Definition at line 163 of file milc_interface.cpp.

◆ qudaGaugeFixingFFT()

void qudaGaugeFixingFFT	(	int	precision,
		unsigned int	gauge_dir,
		int	Nsteps,
		int	verbose_interval,
		double	alpha,
		unsigned int	autotune,
		double	tolerance,
		unsigned int	stopWtheta,
		void *	milc_sitelink
	)

Gauge fixing with Steepest descent method with FFTs with support for single GPU only.

Parameters

[in]	precision,1	for single precision else for double precision
[in]	gauge_dir,3	for Coulomb gauge fixing, other for Landau gauge fixing
[in]	Nsteps,maximum	number of steps to perform gauge fixing
[in]	verbose_interval,print	gauge fixing info when iteration count is a multiple of this
[in]	alpha,gauge	fixing parameter of the method, most common value is 0.08
[in]	autotune,1	to autotune the method, i.e., if the Fg inverts its tendency we decrease the alpha value
[in]	tolerance,torelance	value to stop the method, if this value is zero then the method stops when iteration reachs the maximum number of steps defined by Nsteps
[in]	stopWtheta,0	for MILC criterium and 1 to use the theta value
[in,out]	milc_sitelink,MILC	gauge field to be fixed

Definition at line 2730 of file milc_interface.cpp.

◆ qudaGaugeFixingOVR()

void qudaGaugeFixingOVR	(	const int	precision,
		const unsigned int	gauge_dir,
		const int	Nsteps,
		const int	verbose_interval,
		const double	relax_boost,
		const double	tolerance,
		const unsigned int	reunit_interval,
		const unsigned int	stopWtheta,
		void *	milc_sitelink
	)

Gauge fixing with overrelaxation with support for single and multi GPU.

Parameters

[in]	precision,1	for single precision else for double precision
[in]	gauge_dir,3	for Coulomb gauge fixing, other for Landau gauge fixing
[in]	Nsteps,maximum	number of steps to perform gauge fixing
[in]	verbose_interval,print	gauge fixing info when iteration count is a multiple of this
[in]	relax_boost,gauge	fixing parameter of the overrelaxation method, most common value is 1.5 or 1.7.
[in]	tolerance,torelance	value to stop the method, if this value is zero then the method stops when iteration reachs the maximum number of steps defined by Nsteps
[in]	reunit_interval,reunitarize	gauge field when iteration count is a multiple of this
[in]	stopWtheta,0	for MILC criterium and 1 to use the theta value
[in,out]	milc_sitelink,MILC	gauge field to be fixed

Definition at line 2711 of file milc_interface.cpp.

◆ qudaGaugeForce()

void qudaGaugeForce	(	int	precision,
		int	num_loop_types,
		double	milc_loop_coeff[3],
		double	eb3,
		QudaMILCSiteArg_t *	arg
	)

Compute the gauge force and update the momentum field. All fields here are CPU fields in MILC order, and their precisions should match.

Parameters

precision	The precision of the field (2 - double, 1 - single)
num_loop_types	1, 2 or 3
milc_loop_coeff	Coefficients of the different loops in the Symanzik action
eb3	The integration step size (for MILC this is dt*beta/3)
arg	Metadata for MILC's internal site struct array

Definition at line 655 of file milc_interface.cpp.

◆ qudaGaugeForcePhased()

void qudaGaugeForcePhased	(	int	precision,
		int	num_loop_types,
		double	milc_loop_coeff[3],
		double	eb3,
		QudaMILCSiteArg_t *	arg,
		int	phase_in
	)

Compute the gauge force and update the momentum field. All fields here are CPU fields in MILC order, and their precisions should match.

Parameters

precision	The precision of the field (2 - double, 1 - single)
num_loop_types	1, 2 or 3
milc_loop_coeff	Coefficients of the different loops in the Symanzik action
eb3	The integration step size (for MILC this is dt*beta/3)
arg	Metadata for MILC's internal site struct array
phase_in	whether staggered phases are applied

Definition at line 550 of file milc_interface.cpp.

◆ qudaHisqForce()

void qudaHisqForce	(	int	precision,
		int	num_terms,
		int	num_naik_terms,
		double	dt,
		double **	coeff,
		void **	quark_field,
		const double	level2_coeff[6],
		const double	fat7_coeff[6],
		const void *const	w_link,
		const void *const	v_link,
		const void *const	u_link,
		void *const	milc_momentum
	)

Compute the fermion force for the HISQ quark action. All fields are host fields in MILC order, and the precision of these fields must match.

Parameters

precision	The precision of the fields
num_terms	The number of quark fields
num_naik_terms	The number of naik contributions
dt	Integrating step size
coeff	The coefficients multiplying the fermion fields in the outer product
quark_field	The input fermion field.
level2_coeff	The coefficients for the second level of smearing in the quark action.
fat7_coeff	The coefficients for the first level of smearing (fat7) in the quark action.
w_link	Unitarized link variables obtained by applying fat7 smearing and unitarization to the original links.
v_link	Fat7 link variables.
u_link	SU(3) think link variables.
milc_momentum	The momentum contribution from the quark action.

Definition at line 283 of file milc_interface.cpp.

◆ qudaHisqParamsInit()

void qudaHisqParamsInit ( QudaHisqParams_t hisq_params )

Set the algorithms to use for HISQ fermion calculations, e.g., SVD parameters for reunitarization.

Parameters

hisq_params Meta data desribing the algorithms to use for HISQ fermions

Definition at line 169 of file milc_interface.cpp.

◆ qudaInit()

void qudaInit ( QudaInitArgs_t input )

Initialize the QUDA context.

Parameters

input Meta data for the QUDA context

Definition at line 96 of file milc_interface.cpp.

◆ qudaInvert()

void qudaInvert	(	int	external_precision,
		int	quda_precision,
		double	mass,
		QudaInvertArgs_t	inv_args,
		double	target_residual,
		double	target_fermilab_residual,
		const void *const	milc_fatlink,
		const void *const	milc_longlink,
		void *	source,
		void *	solution,
		double *const	final_resid,
		double *const	final_rel_resid,
		int *	num_iters
	)

Solve Ax=b for an improved staggered operator. All fields are fields passed and returned are host (CPU) field in MILC order. This function requires that persistent gauge and clover fields have been created prior. This interface is experimental.

Parameters

external_precision	Precision of host fields passed to QUDA (2 - double, 1 - single)
quda_precision	Precision for QUDA to use (2 - double, 1 - single)
mass	Fermion mass parameter
inv_args	Struct setting some solver metadata
target_residual	Target residual
target_relative_residual	Target Fermilab residual
milc_fatlink	Fat-link field on the host
milc_longlink	Long-link field on the host
source	Right-hand side source field
solution	Solution spinor field
final_residual	True residual
final_relative_residual	True Fermilab residual
num_iters	Number of iterations taken

Definition at line 985 of file milc_interface.cpp.

◆ qudaInvertMG()

void qudaInvertMG	(	int	external_precision,
		int	quda_precision,
		double	mass,
		QudaInvertArgs_t	inv_args,
		double	target_residual,
		double	target_fermilab_residual,
		const void *const	milc_fatlink,
		const void *const	milc_longlink,
		void *	mg_pack_ptr,
		int	mg_rebuild_type,
		void *	source,
		void *	solution,
		double *const	final_residual,
		double *const	final_fermilab_residual,
		int *	num_iters
	)

Solve Ax=b for an improved staggered operator using MG. All fields are fields passed and returned are host (CPU) field in MILC order. This function requires that persistent gauge and clover fields have been created prior. It also requires a multigrid parameter built from qudaSetupMultigrid This interface is experimental.

Parameters

external_precision	Precision of host fields passed to QUDA (2 - double, 1 - single)
quda_precision	Precision for QUDA to use (2 - double, 1 - single)
mass	Fermion mass parameter
inv_args	Struct setting some solver metadata
target_residual	Target residual
target_relative_residual	Target Fermilab residual
milc_fatlink	Fat-link field on the host
milc_longlink	Long-link field on the host
mg_pack_ptr	MG preconditioner structure created by qudaSetupMultigrid
mg_rebuild_type	whether to do a full (1) or thin (0) MG rebuild
source	Right-hand side source field
solution	Solution spinor field
final_residual	True residual
final_relative_residual	True Fermilab residual
num_iters	Number of iterations taken

Definition at line 2134 of file milc_interface.cpp.

◆ qudaInvertMsrc()

void qudaInvertMsrc	(	int	external_precision,
		int	quda_precision,
		double	mass,
		QudaInvertArgs_t	inv_args,
		double	target_residual,
		double	target_fermilab_residual,
		const void *const	fatlink,
		const void *const	longlink,
		void **	sourceArray,
		void **	solutionArray,
		double *const	final_residual,
		double *const	final_fermilab_residual,
		int *	num_iters,
		int	num_src
	)

Solve Ax=b for an improved staggered operator with many right hand sides. All fields are fields passed and returned are host (CPU) field in MILC order. This function requires that persistent gauge and clover fields have been created prior. This interface is experimental.

Parameters

external_precision	Precision of host fields passed to QUDA (2 - double, 1 - single)
quda_precision	Precision for QUDA to use (2 - double, 1 - single)
mass	Fermion mass parameter
inv_args	Struct setting some solver metadata
target_residual	Target residual
target_relative_residual	Target Fermilab residual
milc_fatlink	Fat-link field on the host
milc_longlink	Long-link field on the host
source	array of right-hand side source fields
solution	array of solution spinor fields
final_residual	True residual
final_relative_residual	True Fermilab residual
num_iters	Number of iterations taken
num_src	Number of source fields

Definition at line 1116 of file milc_interface.cpp.

◆ qudaLoadCloverField()

void qudaLoadCloverField	(	int	external_precision,
		int	quda_precision,
		QudaInvertArgs_t	inv_args,
		void *	milc_clover,
		void *	milc_clover_inv,
		QudaSolutionType	solution_type,
		QudaSolveType	solve_type,
		QudaInverterType	inverter,
		double	clover_coeff,
		int	compute_trlog,
		double *	trlog
	)

Definition at line 2441 of file milc_interface.cpp.

◆ qudaLoadGaugeField()

void qudaLoadGaugeField	(	int	external_precision,
		int	quda_precision,
		QudaInvertArgs_t	inv_args,
		const void *	milc_link
	)

Load the gauge field from the host.

Parameters

external_precision	Precision of host fields passed to QUDA (2 - double, 1 - single)
quda_precision	Precision for QUDA to use (2 - double, 1 - single)
inv_args	Meta data
milc_link	Base pointer to host gauge field (regardless of dimensionality)

Definition at line 2422 of file milc_interface.cpp.

◆ qudaLoadKSLink()

void qudaLoadKSLink	(	int	precision,
		QudaFatLinkArgs_t	fatlink_args,
		const double	act_path_coeff[6],
		void *	inlink,
		void *	fatlink,
		void *	longlink
	)

Compute the fat and long links using the input gauge field. All fields passed here are host fields, that must be preallocated. The precision of all fields must match.

Parameters

precision	The precision of the fields
fatlink_args	Meta data for the algorithms to deploy
act_path_coeff	Array of coefficients for each path in the action
inlink	Host gauge field used for input
fatlink	Host fat-link field that is computed
longlink	Host long-link field that is computed

Definition at line 239 of file milc_interface.cpp.

◆ qudaLoadUnitarizedLink()

void qudaLoadUnitarizedLink	(	int	precision,
		QudaFatLinkArgs_t	fatlink_args,
		const double	path_coeff[6],
		void *	inlink,
		void *	fatlink,
		void *	ulink
	)

Compute the fat links and unitzarize using the input gauge field. All fields passed here are host fields, that must be preallocated. The precision of all fields must match.

Parameters

precision	The precision of the fields
fatlink_args	Meta data for the algorithms to deploy
path_coeff	Array of coefficients for each path in the action
inlink	Host gauge field used for input
fatlink	Host fat-link field that is computed
ulink	Host unitarized field that is computed

Definition at line 263 of file milc_interface.cpp.

◆ qudamilc_called() [1/2]

template<bool start>

void qudamilc_called ( const char * func )

inline

Definition at line 92 of file milc_interface.cpp.

◆ qudamilc_called() [2/2]

template<bool start>

void qudamilc_called	(	const char *	func,
		QudaVerbosity	verb
	)

inline

Definition at line 72 of file milc_interface.cpp.

◆ qudaMomAction()

double qudaMomAction	(	int	precision,
		QudaMILCSiteArg_t *	arg
	)

Evaluate the momentum contribution to the Hybrid Monte Carlo action. MILC convention is applied, subtracting 4.0 from each momentum matrix to increase stability.

Parameters

precision	Precision of the field (2 - double, 1 - single)
arg	Metadata for MILC's internal site struct array

Returns: momentum action

Definition at line 467 of file milc_interface.cpp.

◆ qudaMomLoad()

void qudaMomLoad	(	int	precision,
		QudaMILCSiteArg_t *	arg
	)

Download the momentum from MILC and place into QUDA's resident momentum field. The source momentum field can either be as part of a MILC site struct (QUDA_MILC_SITE_GAUGE_ORDER) or as a separate field (QUDA_MILC_GAUGE_ORDER).

Parameters

precision	Precision of the field (2 - double, 1 - single)
arg	Metadata for MILC's internal site struct array

Definition at line 426 of file milc_interface.cpp.

◆ qudaMomSave()

void qudaMomSave	(	int	precision,
		QudaMILCSiteArg_t *	arg
	)

Upload the momentum to MILC from QUDA's resident momentum field. The destination momentum field can either be as part of a MILC site struct (QUDA_MILC_SITE_GAUGE_ORDER) or as a separate field (QUDA_MILC_GAUGE_ORDER).

Parameters

precision	Precision of the field (2 - double, 1 - single)
arg	Metadata for MILC's internal site struct array

Definition at line 447 of file milc_interface.cpp.

◆ qudaMultigridCreate()

void* qudaMultigridCreate	(	int	external_precision,
		int	quda_precision,
		double	mass,
		QudaInvertArgs_t	inv_args,
		const void *const	milc_fatlink,
		const void *const	milc_longlink,
		const char *const	mg_param_file
	)

Prepare a staggered/HISQ multigrid solve with given fat and long links. All fields passed are host (CPU) fields in MILC order. This function requires persistent gauge fields. This interface is experimental.

Parameters

external_precision	Precision of host fields passed to QUDA (2 - double, 1 - single)
quda_precision	Precision for QUDA to use (2 - double, 1 - single)
mass	Fermion mass parameter
inv_args	Struct setting some solver metadata; required for tadpole, naik coeff
milc_fatlink	Fat-link field on the host
milc_longlink	Long-link field on the host
mg_param_file	Path to an input text file describing the MG solve, to be documented on QUDA wiki

Returns: Void pointer wrapping a pack of multigrid-related structures

Definition at line 2073 of file milc_interface.cpp.

◆ qudaMultigridDestroy()

void qudaMultigridDestroy ( void * mg_pack_ptr )

Clean up a staggered/HISQ multigrid object, freeing all internal fields and otherwise allocated memory.

Parameters

mg_pack_ptr Void pointer mapping to the multigrid structure returned by qudaSetupMultigrid

Definition at line 2255 of file milc_interface.cpp.

◆ qudaMultishiftInvert()

void qudaMultishiftInvert	(	int	external_precision,
		int	quda_precision,
		int	num_offsets,
		double *const	offset,
		QudaInvertArgs_t	inv_args,
		const double	target_residual[],
		const double	target_fermilab_residual[],
		const void *const	fatlink,
		const void *const	longlink,
		void *	source,
		void **	solutionArray,
		double *const	final_residual,
		double *const	final_fermilab_residual,
		int *	num_iters
	)

Definition at line 884 of file milc_interface.cpp.

◆ qudaRephase()

void qudaRephase	(	int	prec,
		void *	gauge,
		int	flag,
		double	i_mu
	)

Apply the staggered phase factors to the gauge field. If the imaginary chemical potential is non-zero then the phase factor exp(imu/T) will be applied to the links in the temporal direction.

Parameters

prec	Precision of the gauge field
gauge_h	The gauge field
flag	Whether to apply to remove the staggered phase
i_mu	Imaginary chemical potential

Definition at line 376 of file milc_interface.cpp.

◆ qudaSaveGaugeField()

void qudaSaveGaugeField	(	void *	gauge,
		void *	inGauge
	)

Copy the QUDA gauge (matrix) field on the device to the CPU

Parameters

outGauge	Pointer to the host gauge field
inGauge	Pointer to the device gauge field (QUDA device field)

Definition at line 2282 of file milc_interface.cpp.

◆ qudaSetLayout()

void qudaSetLayout ( QudaLayout_t layout )

Set set the local dimensions and machine topology for QUDA to use

Parameters

layout Struct defining local dimensions and machine topology

Definition at line 129 of file milc_interface.cpp.

◆ qudaSetMPICommHandle()

void qudaSetMPICommHandle ( void * mycomm )

Optional: Set the MPI Comm Handle if it is not MPI_COMM_WORLD

Parameters

input Pointer to an MPI_Comm handle, static cast as a void *

Definition at line 94 of file milc_interface.cpp.

◆ qudaUnitarizeSU3()

void qudaUnitarizeSU3	(	int	prec,
		double	tol,
		QudaMILCSiteArg_t *	arg
	)

Project the input field on the SU(3) group. If the target tolerance is not met, this routine will give a runtime error.

Parameters

prec	Precision of the gauge field
tol	The tolerance to which we iterate
arg	Metadata for MILC's internal site struct array

Definition at line 423 of file milc_interface.cpp.

◆ qudaUnitarizeSU3Phased()

void qudaUnitarizeSU3Phased	(	int	prec,
		double	tol,
		QudaMILCSiteArg_t *	arg,
		int	phase_in
	)

Project the input field on the SU(3) group. If the target tolerance is not met, this routine will give a runtime error.

Parameters

prec	Precision of the gauge field
tol	The tolerance to which we iterate
arg	Metadata for MILC's internal site struct array
phase_in	whether staggered phases are applied

Definition at line 392 of file milc_interface.cpp.

◆ qudaUpdateU()

void qudaUpdateU	(	int	precision,
		double	eps,
		QudaMILCSiteArg_t *	arg
	)

Evolve the gauge field by step size dt, using the momentum field I.e., Evalulate U(t+dt) = e(dt pi) U(t). All fields are CPU fields in MILC order.

Parameters

precision	Precision of the field (2 - double, 1 - single)
dt	The integration step size step
arg	Metadata for MILC's internal site struct array

Definition at line 374 of file milc_interface.cpp.

◆ qudaUpdateUPhased()

void qudaUpdateUPhased	(	int	precision,
		double	eps,
		QudaMILCSiteArg_t *	arg,
		int	phase_in
	)

Evolve the gauge field by step size dt, using the momentum field I.e., Evalulate U(t+dt) = e(dt pi) U(t). All fields are CPU fields in MILC order.

Parameters

precision	Precision of the field (2 - double, 1 - single)
dt	The integration step size step
arg	Metadata for MILC's internal site struct array
phase_in	whether staggered phases are applied

Definition at line 369 of file milc_interface.cpp.

◆ qudaUpdateUPhasedPipeline()

void qudaUpdateUPhasedPipeline	(	int	precision,
		double	eps,
		QudaMILCSiteArg_t *	arg,
		int	phase_in,
		int	want_gaugepipe
	)

Evolve the gauge field by step size dt, using the momentum field I.e., Evalulate U(t+dt) = e(dt pi) U(t). All fields are CPU fields in MILC order.

Parameters

precision	Precision of the field (2 - double, 1 - single)
dt	The integration step size step
arg	Metadata for MILC's internal site struct array
phase_in	whether staggered phases are applied
want_gaugepipe	whether to enabled QUDA gaugepipe for HMC

Definition at line 328 of file milc_interface.cpp.

◆ setDeflationParam()

void setDeflationParam	(	QudaPrecision	ritz_prec,
		QudaFieldLocation	location_ritz,
		QudaMemoryType	mem_type_ritz,
		QudaExtLibType	deflation_ext_lib,
		char	vec_infile[],
		char	vec_outfile[],
		QudaEigParam *	df_param
	)

Definition at line 851 of file milc_interface.cpp.

◆ setGaugeParams()

void setGaugeParams	(	QudaGaugeParam &	qudaGaugeParam,
		const int	dim[4],
		QudaInvertArgs_t &	inv_args,
		int	external_precision,
		int	quda_precision
	)

Definition at line 2332 of file milc_interface.cpp.

◆ setInvertParam()

void setInvertParam	(	QudaInvertParam &	invertParam,
		QudaInvertArgs_t &	inv_args,
		int	external_precision,
		int	quda_precision,
		double	kappa,
		double	reliable_delta
	)

Definition at line 2379 of file milc_interface.cpp.

Classes

Macros

Functions

Macro Definition Documentation

◆ MAX

◆ POP_RANGE

◆ PUSH_RANGE

◆ QUDAMILC_VERBOSE

Function Documentation

◆ milcSetMultigridEigParam()

◆ milcSetMultigridParam()

◆ qudaAllocateManaged()

◆ qudaAllocatePinned()

◆ qudaAsqtadForce()

◆ qudaCloverForce()

◆ qudaCloverInvert()

◆ qudaCloverMultishiftInvert()

◆ qudaComputeOprod()

◆ qudaCreateGaugeField()

◆ qudaDestroyGaugeField()

◆ qudaDslash()

◆ qudaEigCGCloverInvert()

◆ qudaEigCGInvert()

◆ qudaFinalize()

◆ qudaFreeCloverField()

◆ qudaFreeGaugeField()

◆ qudaFreeManaged()

◆ qudaFreePinned()

◆ qudaGaugeFixingFFT()

◆ qudaGaugeFixingOVR()

◆ qudaGaugeForce()

◆ qudaGaugeForcePhased()

◆ qudaHisqForce()

◆ qudaHisqParamsInit()

◆ qudaInit()

◆ qudaInvert()

◆ qudaInvertMG()

◆ qudaInvertMsrc()

◆ qudaLoadCloverField()

◆ qudaLoadGaugeField()

◆ qudaLoadKSLink()

◆ qudaLoadUnitarizedLink()

◆ qudamilc_called() [1/2]

◆ qudamilc_called() [2/2]

◆ qudaMomAction()

◆ qudaMomLoad()

◆ qudaMomSave()

◆ qudaMultigridCreate()

◆ qudaMultigridDestroy()

◆ qudaMultishiftInvert()

◆ qudaRephase()

◆ qudaSaveGaugeField()

◆ qudaSetLayout()

◆ qudaSetMPICommHandle()

◆ qudaUnitarizeSU3()

◆ qudaUnitarizeSU3Phased()

◆ qudaUpdateU()

◆ qudaUpdateUPhased()

◆ qudaUpdateUPhasedPipeline()

◆ setDeflationParam()

◆ setGaugeParams()

◆ setInvertParam()