QUDA
v1.1.0
A library for QCD on GPUs
|
#include <dirac_quda.h>
Public Member Functions | |
DiracStaggeredKD (const DiracParam ¶m) | |
DiracStaggeredKD (const DiracStaggeredKD &dirac) | |
virtual | ~DiracStaggeredKD () |
DiracStaggeredKD & | operator= (const DiracStaggeredKD &dirac) |
virtual void | checkParitySpinor (const ColorSpinorField &, const ColorSpinorField &) const |
Check parity spinors are usable (check geometry ?) More... | |
virtual bool | hasDslash () const |
Whether or not the operator has a single-parity Dslash. More... | |
virtual void | Dslash (ColorSpinorField &out, const ColorSpinorField &in, const QudaParity parity) const |
apply 'dslash' operator for the DiracOp. This may be e.g. AD More... | |
virtual void | DslashXpay (ColorSpinorField &out, const ColorSpinorField &in, const QudaParity parity, const ColorSpinorField &x, const double &k) const |
Xpay version of Dslash. More... | |
virtual void | M (ColorSpinorField &out, const ColorSpinorField &in) const |
Apply M for the dirac op. E.g. the Schur Complement operator. More... | |
virtual void | MdagM (ColorSpinorField &out, const ColorSpinorField &in) const |
Apply MdagM operator which may be optimized. More... | |
void | KahlerDiracInv (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 |
virtual void | prepareSpecialMG (ColorSpinorField *&src, ColorSpinorField *&sol, ColorSpinorField &x, ColorSpinorField &b, const QudaSolutionType solType) const |
virtual void | reconstructSpecialMG (ColorSpinorField &x, const ColorSpinorField &b, const QudaSolutionType solType) const |
virtual bool | hasSpecialMG () const |
specifies whether or not there's a specialized prepare/reconstruct used before/after transfering to/from the coarse level in MG More... | |
virtual QudaDiracType | getDiracType () const |
returns the Dirac type More... | |
virtual void | updateFields (cudaGaugeField *gauge_in, cudaGaugeField *fat_gauge_in, cudaGaugeField *long_gauge_in, cudaCloverField *clover_in) |
Update the internal gauge, fat gauge, long gauge, clover field pointer as appropriate. These are pointers as opposed to references to support passing in nullptr . More... | |
void | createCoarseOp (GaugeField &Y, GaugeField &X, const Transfer &T, double kappa, double mass, double mu=0., double mu_factor=0.) const |
Create the coarse staggered KD operator. More... | |
virtual void | prefetch (QudaFieldLocation mem_space, qudaStream_t stream=0) const |
If managed memory and prefetch is enabled, prefetch all relevant memory fields (gauge, clover, temporary spinors). Will only grab the inverse clover unless the clover field is needed for asymmetric preconditioning to the CPU or GPU as requested. More... | |
Public Member Functions inherited from quda::DiracStaggered | |
DiracStaggered (const DiracParam ¶m) | |
DiracStaggered (const DiracStaggered &dirac) | |
virtual | ~DiracStaggered () |
DiracStaggered & | operator= (const DiracStaggered &dirac) |
virtual const cudaGaugeField * | getGaugeField () const |
Get the fine gauge field for MG setup. More... | |
Public Member Functions inherited from quda::Dirac | |
Dirac (const DiracParam ¶m) | |
Dirac (const Dirac &dirac) | |
virtual | ~Dirac () |
Dirac & | operator= (const Dirac &dirac) |
void | setCommDim (const int commDim_[QUDA_MAX_DIM]) const |
Enable / disable communications for the Dirac operator. More... | |
virtual bool | isCoarse () const |
Whether the Dirac object is the DiracCoarse. More... | |
virtual void | checkFullSpinor (const ColorSpinorField &, const ColorSpinorField &) const |
check full spinors are compatible (check geometry ?) More... | |
void | checkSpinorAlias (const ColorSpinorField &, const ColorSpinorField &) const |
check spinors do not alias More... | |
virtual void | MdagMLocal (ColorSpinorField &out, const ColorSpinorField &in) const |
Apply the local MdagM operator: equivalent to applying zero Dirichlet boundary condition to MdagM on each rank. Depending on the number of stencil steps of the fermion type, this may require additional effort to include the terms that hop out of the boundary and then hop back. More... | |
virtual void | Dslash4 (ColorSpinorField &out, const ColorSpinorField &in, const QudaParity parity) const |
Apply the local MdagM operator: equivalent to applying zero Dirichlet boundary condition to MdagM on each rank. Depending on the number of stencil steps of the fermion type, this may require additional effort to include the terms that hop out of the boundary and then hop back. More... | |
void | Mdag (ColorSpinorField &out, const ColorSpinorField &in) const |
Apply Mdag (daggered operator of M. More... | |
void | MMdag (ColorSpinorField &out, const ColorSpinorField &in) const |
Apply Normal Operator. More... | |
void | setMass (double mass) |
double | Kappa () const |
accessor for Kappa (mass parameter) More... | |
virtual double | Mass () const |
accessor for Mass (in case of a factor of 2 for staggered) More... | |
virtual double | Mu () const |
accessor for twist parameter – overrride can return better value More... | |
virtual double | MuFactor () const |
accessor for mu factoo for MG/ – override can return a better value More... | |
unsigned long long | Flops () const |
returns and then zeroes flopcount More... | |
QudaMatPCType | getMatPCType () const |
returns preconditioning type More... | |
int | getStencilSteps () const |
I have no idea what this does. More... | |
void | Dagger (QudaDagType dag) const |
sets whether operator is daggered or not More... | |
void | flipDagger () const |
Flips value of daggered. More... | |
virtual bool | hermitian () const |
is operator hermitian More... | |
QudaPrecision | HaloPrecision () const |
void | setHaloPrecision (QudaPrecision halo_precision_) const |
Public Member Functions inherited from quda::Object | |
Object () | |
virtual | ~Object () |
Protected Attributes | |
cudaGaugeField * | Xinv |
Protected Attributes inherited from quda::Dirac | |
cudaGaugeField * | gauge |
double | kappa |
double | mass |
int | laplace3D |
QudaMatPCType | matpcType |
QudaDagType | dagger |
unsigned long long | flops |
ColorSpinorField * | tmp1 |
ColorSpinorField * | tmp2 |
QudaDiracType | type |
QudaPrecision | halo_precision |
int | commDim [QUDA_MAX_DIM] |
TimeProfile | profile |
Additional Inherited Members | |
Static Public Member Functions inherited from quda::Dirac | |
static Dirac * | create (const DiracParam ¶m) |
Creates a subclass from parameters. More... | |
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) |
Protected Member Functions inherited from quda::Dirac | |
bool | newTmp (ColorSpinorField **, const ColorSpinorField &) const |
void | deleteTmp (ColorSpinorField **, const bool &reset) const |
Definition at line 1269 of file dirac_quda.h.
quda::DiracStaggeredKD::DiracStaggeredKD | ( | const DiracParam & | param | ) |
inverse Kahler-Dirac matrix
Definition at line 9 of file dirac_staggered_kd.cpp.
quda::DiracStaggeredKD::DiracStaggeredKD | ( | const DiracStaggeredKD & | dirac | ) |
Definition at line 11 of file dirac_staggered_kd.cpp.
|
virtual |
Definition at line 13 of file dirac_staggered_kd.cpp.
|
virtual |
Check parity spinors are usable (check geometry ?)
Reimplemented from quda::DiracStaggered.
Definition at line 24 of file dirac_staggered_kd.cpp.
|
virtual |
Create the coarse staggered KD operator.
Takes the multigrid transfer class, which knows about the coarse grid blocking, as well as having prolongate and restrict member functions, and returns color matrices Y[0..2*dim-1] corresponding to the coarse grid hopping terms and X corresponding to the coarse grid "clover" term.
T[in] | Transfer operator defining the coarse grid |
Y[out] | Coarse link field |
X[out] | Coarse clover field |
kappa | Kappa parameter for the coarse operator (ignored, set to 1.0) |
mass | Mass parameter for the coarse operator (gets explicitly built into clover) |
mu | Mu parameter for the coarse operator (ignored for staggered) |
mu_factor | Mu scaling factor for the coarse operator (ignored for staggered) |
Reimplemented from quda::DiracStaggered.
Definition at line 222 of file dirac_staggered_kd.cpp.
|
virtual |
apply 'dslash' operator for the DiracOp. This may be e.g. AD
Reimplemented from quda::DiracStaggered.
Definition at line 41 of file dirac_staggered_kd.cpp.
|
virtual |
Xpay version of Dslash.
Reimplemented from quda::DiracStaggered.
Definition at line 46 of file dirac_staggered_kd.cpp.
|
inlinevirtual |
returns the Dirac type
Reimplemented from quda::DiracStaggered.
Definition at line 1305 of file dirac_quda.h.
|
inlinevirtual |
Whether or not the operator has a single-parity Dslash.
Reimplemented from quda::Dirac.
Definition at line 1284 of file dirac_quda.h.
|
inlinevirtual |
specifies whether or not there's a specialized prepare/reconstruct used before/after transfering to/from the coarse level in MG
Reimplemented from quda::Dirac.
Definition at line 1303 of file dirac_quda.h.
void quda::DiracStaggeredKD::KahlerDiracInv | ( | ColorSpinorField & | out, |
const ColorSpinorField & | in | ||
) | const |
Definition at line 134 of file dirac_staggered_kd.cpp.
|
virtual |
Apply M for the dirac op. E.g. the Schur Complement operator.
Reimplemented from quda::DiracStaggered.
Definition at line 53 of file dirac_staggered_kd.cpp.
|
virtual |
Apply MdagM operator which may be optimized.
Reimplemented from quda::DiracStaggered.
Definition at line 123 of file dirac_staggered_kd.cpp.
DiracStaggeredKD & quda::DiracStaggeredKD::operator= | ( | const DiracStaggeredKD & | dirac | ) |
Definition at line 15 of file dirac_staggered_kd.cpp.
|
virtual |
If managed memory and prefetch is enabled, prefetch all relevant memory fields (gauge, clover, temporary spinors). Will only grab the inverse clover unless the clover field is needed for asymmetric preconditioning to the CPU or GPU as requested.
[in] | mem_space | Memory space we are prefetching to |
[in] | stream | Which stream to run the prefetch in (default 0) |
Reimplemented from quda::Dirac.
Definition at line 232 of file dirac_staggered_kd.cpp.
|
virtual |
Reimplemented from quda::DiracStaggered.
Definition at line 139 of file dirac_staggered_kd.cpp.
|
virtual |
Reimplemented from quda::Dirac.
Definition at line 152 of file dirac_staggered_kd.cpp.
|
virtual |
Reimplemented from quda::DiracStaggered.
Definition at line 182 of file dirac_staggered_kd.cpp.
|
virtual |
Reimplemented from quda::Dirac.
Definition at line 190 of file dirac_staggered_kd.cpp.
|
virtual |
Update the internal gauge, fat gauge, long gauge, clover field pointer as appropriate. These are pointers as opposed to references to support passing in nullptr
.
gauge_in | Updated gauge field |
fat_gauge_in | Updated fat links |
long_gauge_in | Updated long links |
clover_in | Updated clover field |
Reimplemented from quda::Dirac.
Definition at line 213 of file dirac_staggered_kd.cpp.
|
mutableprotected |
Definition at line 1273 of file dirac_quda.h.