#include <dirac_quda.h>

Inheritance diagram for quda::DiracCoarse:

Collaboration diagram for quda::DiracCoarse:

[legend]

Public Member Functions
double	Mu () const

double	MuFactor () const

	DiracCoarse (const DiracParam &param, bool enable_gpu=true)

	DiracCoarse (const DiracParam &param, cpuGaugeField Y_h, cpuGaugeField X_h, cpuGaugeField Xinv_h, cpuGaugeField Yhat_h, cudaGaugeField Y_d=0, cudaGaugeField X_d=0, cudaGaugeField Xinv_d=0, cudaGaugeField Yhat_d=0)

	DiracCoarse (const DiracCoarse &dirac, const DiracParam &param)

virtual	~DiracCoarse ()

void	Clover (ColorSpinorField &out, const ColorSpinorField &in, const QudaParity parity) const
	Apply the coarse clover operator. More...

void	CloverInv (ColorSpinorField &out, const ColorSpinorField &in, const QudaParity parity) const
	Apply the inverse coarse clover operator. More...

virtual void	Dslash (ColorSpinorField &out, const ColorSpinorField &in, const QudaParity parity) const
	Apply DslashXpay out = (D * in) More...

virtual void	DslashXpay (ColorSpinorField &out, const ColorSpinorField &in, const QudaParity parity, const ColorSpinorField &x, const double &k) const
	Apply DslashXpay out = (D * in + A * x) More...

virtual void	M (ColorSpinorField &out, const ColorSpinorField &in) const
	Apply the full operator. More...

virtual void	MdagM (ColorSpinorField &out, const ColorSpinorField &in) const

virtual void	prepare (ColorSpinorField &src, ColorSpinorField &sol, ColorSpinorField &x, ColorSpinorField &b, const QudaSolutionType) const

virtual void	reconstruct (ColorSpinorField &x, const ColorSpinorField &b, const QudaSolutionType) const

void	createCoarseOp (GaugeField &Y, GaugeField &X, GaugeField &Xinv, GaugeField &Yhat, const Transfer &T, double kappa, double mu, double mu_factor=0.) const
	Create the coarse operator from this coarse operator. More...

Public Member Functions inherited from quda::Dirac
	Dirac (const DiracParam &param)

	Dirac (const Dirac &dirac)

virtual	~Dirac ()

Dirac &	operator= (const Dirac &dirac)

virtual void	checkParitySpinor (const ColorSpinorField &, const ColorSpinorField &) const

virtual void	checkFullSpinor (const ColorSpinorField &, const ColorSpinorField &) const

void	checkSpinorAlias (const ColorSpinorField &, const ColorSpinorField &) const

void	Mdag (ColorSpinorField &out, const ColorSpinorField &in) const

void	MMdag (ColorSpinorField &out, const ColorSpinorField &in) const

void	setMass (double mass)

double	Kappa () const

unsigned long long	Flops () const

QudaMatPCType	getMatPCType () const

int	getStencilSteps () const

void	Dagger (QudaDagType dag) const

Public Member Functions inherited from quda::Object
	Object ()

virtual	~Object ()

Protected Member Functions
void	initializeCoarse ()

Protected Member Functions inherited from quda::Dirac
bool	newTmp (ColorSpinorField **, const ColorSpinorField &) const

void	deleteTmp (ColorSpinorField **, const bool &reset) const

Protected Attributes
double	mu

double	mu_factor

const Transfer *	transfer

const Dirac *	dirac

cpuGaugeField *	Y_h

cpuGaugeField *	X_h

cpuGaugeField *	Xinv_h

cpuGaugeField *	Yhat_h

cudaGaugeField *	Y_d

cudaGaugeField *	X_d

cudaGaugeField *	Xinv_d

cudaGaugeField *	Yhat_d

bool	enable_gpu

bool	init

Protected Attributes inherited from quda::Dirac
cudaGaugeField *	gauge

double	kappa

double	mass

QudaMatPCType	matpcType

QudaDagType	dagger

unsigned long long	flops

ColorSpinorField *	tmp1

ColorSpinorField *	tmp2

QudaDiracType	type

QudaTune	tune

int	commDim [QUDA_MAX_DIM]

TimeProfile	profile

Additional Inherited Members
Static Public Member Functions inherited from quda::Dirac
static Dirac *	create (const DiracParam &param)

Static Public Member Functions inherited from quda::Object
static void *	operator new (std::size_t size)

static void	operator delete (void *p)

static void *	operator new[] (std::size_t size)

static void	operator delete[] (void *p)

Detailed Description

This class serves as a front-end to the coarse Dslash operator, similar to the other dslash operators.

Definition at line 740 of file dirac_quda.h.

Constructor & Destructor Documentation

◆ DiracCoarse() [1/3]

quda::DiracCoarse::DiracCoarse	(	const DiracParam &	param,
		bool	enable_gpu = `true`
	)

Parameters

[in]	param	Parameters defining this operator
[in]	enable_gpu	Whether to enable this operator for the GPU

Definition at line 7 of file dirac_coarse.cpp.

References initializeCoarse().

Here is the call graph for this function:

◆ DiracCoarse() [2/3]

quda::DiracCoarse::DiracCoarse	(	const DiracParam &	param,
		cpuGaugeField *	Y_h,
		cpuGaugeField *	X_h,
		cpuGaugeField *	Xinv_h,
		cpuGaugeField *	Yhat_h,
		cudaGaugeField *	Y_d = `0`,
		cudaGaugeField *	X_d = `0`,
		cudaGaugeField *	Xinv_d = `0`,
		cudaGaugeField *	Yhat_d = `0`
	)

Parameters

[in]	param	Parameters defining this operator
[in]	Y_h	CPU coarse link field
[in]	X_h	CPU coarse clover field
[in]	Xinv_h	CPU coarse inverse clover field
[in]	Yhat_h	CPU coarse preconditioned link field
[in]	Y_d	GPU coarse link field
[in]	X_d	GPU coarse clover field
[in]	Xinv_d	GPU coarse inverse clover field
[in]	Yhat_d	GPU coarse preconditioned link field

Definition at line 16 of file dirac_coarse.cpp.

◆ DiracCoarse() [3/3]

quda::DiracCoarse::DiracCoarse	(	const DiracCoarse &	dirac,
		const DiracParam &	param
	)

Parameters

[in]	dirac	Another operator instance to clone from (shallow copy)
[in]	param	Parameters defining this operator

Definition at line 27 of file dirac_coarse.cpp.

◆ ~DiracCoarse()

quda::DiracCoarse::~DiracCoarse ( )

virtual

Definition at line 36 of file dirac_coarse.cpp.

References init, X_d, X_h, Xinv_d, Xinv_h, Y_d, Y_h, Yhat_d, and Yhat_h.

Member Function Documentation

◆ Clover()

void quda::DiracCoarse::Clover	(	ColorSpinorField &	out,
		const ColorSpinorField &	in,
		const QudaParity	parity
	)		const

Apply the coarse clover operator.

Parameters

[out]	out	Output field
[in]	in	Input field
	[paraity]	parity Parity which we are applying the operator to

Definition at line 132 of file dirac_coarse.cpp.

References quda::ApplyCoarse(), checkLocation, quda::Dirac::dagger, enable_gpu, errorQuda, quda::Dirac::flops, in, quda::Dirac::kappa, n, out, parity, QUDA_CPU_FIELD_LOCATION, QUDA_CUDA_FIELD_LOCATION, X_d, X_h, Y_d, and Y_h.

Referenced by benchmark(), and quda::DiracCoarsePC::M().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ CloverInv()

void quda::DiracCoarse::CloverInv	(	ColorSpinorField &	out,
		const ColorSpinorField &	in,
		const QudaParity	parity
	)		const

Apply the inverse coarse clover operator.

Parameters

[out]	out	Output field
[in]	in	Input field
	[paraity]	parity Parity which we are applying the operator to

Definition at line 145 of file dirac_coarse.cpp.

References quda::ApplyCoarse(), checkLocation, quda::Dirac::dagger, enable_gpu, errorQuda, quda::Dirac::flops, in, quda::Dirac::kappa, n, out, parity, QUDA_CPU_FIELD_LOCATION, QUDA_CUDA_FIELD_LOCATION, Xinv_d, Xinv_h, Y_d, and Y_h.

Referenced by quda::DiracCoarsePC::prepare(), and quda::DiracCoarsePC::reconstruct().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ createCoarseOp()

void quda::DiracCoarse::createCoarseOp	(	GaugeField &	Y,
		GaugeField &	X,
		GaugeField &	Xinv,
		GaugeField &	Yhat,
		const Transfer &	T,
		double	kappa,
		double	mu,
		double	mu_factor = `0.`
	)		const

virtual

Create the coarse operator from this coarse operator.

Parameters

T[in]	Transfer operator defining the coarse grid
Y[out]	Coarse link field
X[out]	Coarse clover field
Xinv[out]	Coarse clover inverse field
Yhat[out]	Coarse preconditioned link field
kappa	Kappa parameter for the coarse operator
mu	TM mu parameter for the coarse operator
mu_factor	multiplicative factor for the mu parameter

Reimplemented from quda::Dirac.

Reimplemented in quda::DiracCoarsePC.

Definition at line 229 of file dirac_coarse.cpp.

References a, checkLocation, quda::CoarseCoarseOp(), quda::Dirac::kappa, mu, mu_factor, QUDA_COARSE_DIRAC, QUDA_CPU_FIELD_LOCATION, QUDA_MATPC_INVALID, quda::ColorSpinorField::TwistFlavor(), quda::Transfer::Vectors(), X, X_d, X_h, Xinv_d, Xinv_h, Y_d, and Y_h.

Here is the call graph for this function:

◆ Dslash()

void quda::DiracCoarse::Dslash	(	ColorSpinorField &	out,
		const ColorSpinorField &	in,
		const QudaParity	parity
	)		const

virtual

Apply DslashXpay out = (D * in)

Parameters

[out]	out	Output field
[in]	in	Input field
	[paraity]	parity Parity which we are applying the operator to

Implements quda::Dirac.

Reimplemented in quda::DiracCoarsePC.

Definition at line 158 of file dirac_coarse.cpp.

References quda::ApplyCoarse(), checkLocation, quda::Dirac::dagger, enable_gpu, errorQuda, quda::Dirac::flops, in, quda::Dirac::kappa, n, out, parity, QUDA_CPU_FIELD_LOCATION, QUDA_CUDA_FIELD_LOCATION, X_d, X_h, Y_d, and Y_h.

Referenced by benchmark(), quda::DiracCoarsePC::M(), quda::DiracCoarsePC::prepare(), and quda::DiracCoarsePC::reconstruct().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ DslashXpay()

void quda::DiracCoarse::DslashXpay	(	ColorSpinorField &	out,
		const ColorSpinorField &	in,
		const QudaParity	parity,
		const ColorSpinorField &	x,
		const double &	k
	)		const

virtual

Apply DslashXpay out = (D * in + A * x)

Parameters

[out]	out	Output field
[in]	in	Input field
	[paraity]	parity Parity which we are applying the operator to

Implements quda::Dirac.

Reimplemented in quda::DiracCoarsePC.

Definition at line 171 of file dirac_coarse.cpp.

References quda::ApplyCoarse(), checkLocation, quda::Dirac::dagger, enable_gpu, errorQuda, quda::Dirac::flops, in, quda::Dirac::kappa, n, out, parity, QUDA_CPU_FIELD_LOCATION, QUDA_CUDA_FIELD_LOCATION, x, X_d, X_h, Y_d, and Y_h.

Here is the call graph for this function:

◆ initializeCoarse()

void quda::DiracCoarse::initializeCoarse ( )

protected

GPU copy of the preconditioned coarse link field

Definition at line 50 of file dirac_coarse.cpp.

References quda::cudaGaugeField::copy(), quda::cpuGaugeField::copy(), quda::Dirac::createCoarseOp(), dirac, enable_gpu, quda::Transfer::Geo_bs(), gParam, fused_exterior_ndeg_tm_dslash_cuda_gen::i, quda::Dirac::kappa, memcpy(), Mu(), MuFactor(), ndim, quda::ColorSpinorField::Ndim(), quda::ColorSpinorField::Nspin(), quda::Transfer::nvec(), prec, quda::LatticeField::Precision(), QUDA_COARSE_GEOMETRY, QUDA_COARSE_LINKS, QUDA_FLOAT2_GAUGE_ORDER, QUDA_FULL_SITE_SUBSET, QUDA_GHOST_EXCHANGE_NO, QUDA_GHOST_EXCHANGE_PAD, QUDA_MAX_DIM, QUDA_PERIODIC_T, QUDA_QDP_GAUGE_ORDER, QUDA_RECONSTRUCT_NO, QUDA_SCALAR_GEOMETRY, QUDA_ZERO_FIELD_CREATE, quda::Transfer::Spin_bs(), transfer, quda::Transfer::Vectors(), quda::ColorSpinorField::X(), x, X_d, X_h, Xinv_d, Xinv_h, Y_d, Y_h, Yhat_d, and Yhat_h.

Referenced by DiracCoarse().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ M()

void quda::DiracCoarse::M	(	ColorSpinorField &	out,
		const ColorSpinorField &	in
	)		const

virtual

Apply the full operator.

Parameters

[out]	out	output vector, out = M * in
[in]	in	input vector

Implements quda::Dirac.

Reimplemented in quda::DiracCoarsePC.

Definition at line 187 of file dirac_coarse.cpp.

References quda::ApplyCoarse(), checkLocation, quda::Dirac::dagger, enable_gpu, errorQuda, quda::Dirac::flops, in, quda::Dirac::kappa, n, out, QUDA_CPU_FIELD_LOCATION, QUDA_CUDA_FIELD_LOCATION, QUDA_INVALID_PARITY, X_d, X_h, Y_d, and Y_h.

Referenced by benchmark(), and MdagM().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ MdagM()

void quda::DiracCoarse::MdagM	(	ColorSpinorField &	out,
		const ColorSpinorField &	in
	)		const

virtual

Implements quda::Dirac.

Reimplemented in quda::DiracCoarsePC.

Definition at line 199 of file dirac_coarse.cpp.

References quda::Dirac::deleteTmp(), errorQuda, in, M(), quda::Dirac::Mdag(), quda::Dirac::newTmp(), out, QUDA_FULL_SITE_SUBSET, quda::ColorSpinorField::SiteSubset(), and quda::Dirac::tmp1.

Here is the call graph for this function:

◆ Mu()

double quda::DiracCoarse::Mu ( ) const

inlinevirtual

Whether this instance did the allocation or not

Reimplemented from quda::Dirac.

Definition at line 764 of file dirac_quda.h.

References mu.

Referenced by initializeCoarse().

Here is the caller graph for this function:

◆ MuFactor()

double quda::DiracCoarse::MuFactor ( ) const

inlinevirtual

Reimplemented from quda::Dirac.

Definition at line 765 of file dirac_quda.h.

References mu_factor.

Referenced by initializeCoarse().

Here is the caller graph for this function:

◆ prepare()

void quda::DiracCoarse::prepare	(	ColorSpinorField *&	src,
		ColorSpinorField *&	sol,
		ColorSpinorField &	x,
		ColorSpinorField &	b,
		const QudaSolutionType	solType
	)		const

virtual

Implements quda::Dirac.

Reimplemented in quda::DiracCoarsePC.

Definition at line 210 of file dirac_coarse.cpp.

References b, errorQuda, QUDA_MATPC_SOLUTION, QUDA_MATPCDAG_MATPC_SOLUTION, src, and x.

◆ reconstruct()

void quda::DiracCoarse::reconstruct	(	ColorSpinorField &	x,
		const ColorSpinorField &	b,
		const QudaSolutionType	solType
	)		const

virtual

Implements quda::Dirac.

Reimplemented in quda::DiracCoarsePC.

Definition at line 222 of file dirac_coarse.cpp.

Member Data Documentation

◆ dirac

const Dirac* quda::DiracCoarse::dirac

protected

restrictor / prolongator defined here

Definition at line 746 of file dirac_quda.h.

Referenced by initializeCoarse().

◆ enable_gpu

bool quda::DiracCoarse::enable_gpu

protected

Initialize the coarse gauge field

Definition at line 760 of file dirac_quda.h.

Referenced by Clover(), CloverInv(), Dslash(), quda::DiracCoarsePC::Dslash(), DslashXpay(), initializeCoarse(), and M().

◆ init

bool quda::DiracCoarse::init

protected

Whether to enable this operator for the GPU

Definition at line 761 of file dirac_quda.h.

Referenced by ~DiracCoarse().

◆ mu

double quda::DiracCoarse::mu

protected

Definition at line 743 of file dirac_quda.h.

Referenced by createCoarseOp(), quda::DiracCoarsePC::createCoarseOp(), and Mu().

◆ mu_factor

double quda::DiracCoarse::mu_factor

protected

Definition at line 744 of file dirac_quda.h.

Referenced by createCoarseOp(), quda::DiracCoarsePC::createCoarseOp(), and MuFactor().

◆ transfer

const Transfer* quda::DiracCoarse::transfer

protected

Definition at line 745 of file dirac_quda.h.

Referenced by initializeCoarse().

◆ X_d

cudaGaugeField* quda::DiracCoarse::X_d

protected

GPU copy of the coarse link field

Definition at line 754 of file dirac_quda.h.

Referenced by Clover(), createCoarseOp(), quda::DiracCoarsePC::createCoarseOp(), Dslash(), quda::DiracCoarsePC::Dslash(), DslashXpay(), initializeCoarse(), M(), and ~DiracCoarse().

◆ X_h

cpuGaugeField* quda::DiracCoarse::X_h

protected

CPU copy of the coarse link field

Definition at line 749 of file dirac_quda.h.

Referenced by Clover(), createCoarseOp(), quda::DiracCoarsePC::createCoarseOp(), Dslash(), quda::DiracCoarsePC::Dslash(), DslashXpay(), initializeCoarse(), M(), and ~DiracCoarse().

◆ Xinv_d

cudaGaugeField* quda::DiracCoarse::Xinv_d

protected

GPU copy of the coarse clover term

Definition at line 755 of file dirac_quda.h.

Referenced by CloverInv(), createCoarseOp(), quda::DiracCoarsePC::createCoarseOp(), initializeCoarse(), and ~DiracCoarse().

◆ Xinv_h

cpuGaugeField* quda::DiracCoarse::Xinv_h

protected

CPU copy of the coarse clover term

Definition at line 750 of file dirac_quda.h.

Referenced by CloverInv(), createCoarseOp(), quda::DiracCoarsePC::createCoarseOp(), initializeCoarse(), and ~DiracCoarse().

◆ Y_d

cudaGaugeField* quda::DiracCoarse::Y_d

protected

CPU copy of the preconditioned coarse link field

Definition at line 753 of file dirac_quda.h.

Referenced by Clover(), CloverInv(), createCoarseOp(), Dslash(), DslashXpay(), initializeCoarse(), M(), and ~DiracCoarse().

◆ Y_h

cpuGaugeField* quda::DiracCoarse::Y_h

protected

Parent Dirac operator

Definition at line 748 of file dirac_quda.h.

Referenced by Clover(), CloverInv(), createCoarseOp(), Dslash(), DslashXpay(), initializeCoarse(), M(), and ~DiracCoarse().

◆ Yhat_d

cudaGaugeField* quda::DiracCoarse::Yhat_d

protected

GPU copy of inverse coarse clover term

Definition at line 756 of file dirac_quda.h.

Referenced by quda::DiracCoarsePC::createCoarseOp(), quda::DiracCoarsePC::Dslash(), initializeCoarse(), and ~DiracCoarse().

◆ Yhat_h

cpuGaugeField* quda::DiracCoarse::Yhat_h

protected

CPU copy of the inverse coarse clover term

Definition at line 751 of file dirac_quda.h.

Referenced by quda::DiracCoarsePC::createCoarseOp(), quda::DiracCoarsePC::Dslash(), initializeCoarse(), and ~DiracCoarse().

The documentation for this class was generated from the following files:

include/dirac_quda.h
lib/dirac_coarse.cpp

Public Member Functions

Protected Member Functions

Protected Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ DiracCoarse() [1/3]

◆ DiracCoarse() [2/3]

◆ DiracCoarse() [3/3]

◆ ~DiracCoarse()

Member Function Documentation

◆ Clover()

◆ CloverInv()

◆ createCoarseOp()

◆ Dslash()

◆ DslashXpay()

◆ initializeCoarse()

◆ M()

◆ MdagM()

◆ Mu()

◆ MuFactor()

◆ prepare()

◆ reconstruct()

Member Data Documentation

◆ dirac

◆ enable_gpu

◆ init

◆ mu

◆ mu_factor

◆ transfer

◆ X_d

◆ X_h

◆ Xinv_d

◆ Xinv_h

◆ Y_d

◆ Y_h

◆ Yhat_d

◆ Yhat_h