Communication-avoiding CG solver. This solver does un-preconditioned CG, running in steps of nKrylov, build up a polynomial in the linear operator of length nKrylov, and then performs a steepest descent minimization on the resulting basis vectors. For now only implemented using the power basis so is only useful as a preconditioner. More...

#include <invert_quda.h>

Inheritance diagram for quda::CACG:

[legend]

Collaboration diagram for quda::CACG:

[legend]

Public Member Functions
	CACG (DiracMatrix &mat, DiracMatrix &matSloppy, SolverParam &param, TimeProfile &profile)

virtual	~CACG ()

void	operator() (ColorSpinorField &out, ColorSpinorField &in)

Public Member Functions inherited from quda::Solver
	Solver (SolverParam &param, TimeProfile &profile)

virtual	~Solver ()

virtual void	blocksolve (ColorSpinorField &out, ColorSpinorField &in)

bool	convergence (double r2, double hq2, double r2_tol, double hq_tol)

bool	convergenceHQ (double r2, double hq2, double r2_tol, double hq_tol)
	Test for HQ solver convergence – ignore L2 residual. More...

bool	convergenceL2 (double r2, double hq2, double r2_tol, double hq_tol)
	Test for L2 solver convergence – ignore HQ residual. More...

void	PrintStats (const char *name, int k, double r2, double b2, double hq2)
	Prints out the running statistics of the solver (requires a verbosity of QUDA_VERBOSE) More...

void	PrintSummary (const char *name, int k, double r2, double b2, double r2_tol, double hq_tol)
	Prints out the summary of the solver convergence (requires a verbosity of QUDA_SUMMARIZE). Assumes SolverParam.true_res and SolverParam.true_res_hq has been set. More...

void	constructDeflationSpace (const ColorSpinorField &meta, const DiracMatrix &mat, bool svd)
	Constructs the deflation space. More...

virtual double	flops () const

Private Member Functions
void	create (ColorSpinorField &b)
	Initiate the fields needed by the solver. More...

void	compute_alpha ()
	Compute the alpha coefficients. More...

void	compute_beta ()
	Compute the beta coefficients. More...

int	reliable (double &rNorm, double &maxrr, int &rUpdate, const double &r2, const double &delta)

Private Attributes
const DiracMatrix &	mat

const DiracMatrix &	matSloppy

bool	init

bool	lambda_init

QudaCABasis	basis

Complex *	Q_AQandg

Complex *	Q_AS

Complex *	alpha

Complex *	beta

ColorSpinorField *	rp

ColorSpinorField *	tmpp

ColorSpinorField *	tmpp2

ColorSpinorField *	tmp_sloppy

ColorSpinorField *	tmp_sloppy2

std::vector< ColorSpinorField * >	S

std::vector< ColorSpinorField * >	AS

std::vector< ColorSpinorField * >	Q

std::vector< ColorSpinorField * >	Qtmp

std::vector< ColorSpinorField * >	AQ

Additional Inherited Members
Static Public Member Functions inherited from quda::Solver
static Solver *	create (SolverParam &param, DiracMatrix &mat, DiracMatrix &matSloppy, DiracMatrix &matPrecon, TimeProfile &profile)

static double	stopping (double tol, double b2, QudaResidualType residual_type)
	Set the solver L2 stopping condition. More...

Public Attributes inherited from quda::Solver
EigenSolver *	eig_solve

bool	deflate_init = false

std::vector< ColorSpinorField * >	defl_tmp1

std::vector< ColorSpinorField * >	defl_tmp2

Protected Attributes inherited from quda::Solver
SolverParam &	param

TimeProfile &	profile

int	node_parity

Detailed Description

Communication-avoiding CG solver. This solver does un-preconditioned CG, running in steps of nKrylov, build up a polynomial in the linear operator of length nKrylov, and then performs a steepest descent minimization on the resulting basis vectors. For now only implemented using the power basis so is only useful as a preconditioner.

Definition at line 891 of file invert_quda.h.

Constructor & Destructor Documentation

◆ CACG()

quda::CACG::CACG	(	DiracMatrix &	mat,
		DiracMatrix &	matSloppy,
		SolverParam &	param,
		TimeProfile &	profile
	)

Definition at line 15 of file inv_ca_cg.cpp.

◆ ~CACG()

quda::CACG::~CACG ( )

virtual

Definition at line 21 of file inv_ca_cg.cpp.

References alpha, AQ, AS, basis, beta, quda::Solver::defl_tmp1, quda::Solver::defl_tmp2, quda::Solver::deflate_init, quda::SolverParam::evecs, init, quda::SolverParam::is_preconditioner, quda::SolverParam::Nkrylov, quda::Solver::param, quda::Solver::profile, Q, Q_AQandg, Q_AS, Qtmp, QUDA_POWER_BASIS, quda::QUDA_PROFILE_FREE, rp, tmp_sloppy, tmp_sloppy2, tmpp, and tmpp2.

Member Function Documentation

◆ compute_alpha()

void quda::CACG::compute_alpha ( )

private

Compute the alpha coefficients.

Definition at line 302 of file inv_ca_cg.cpp.

References alpha, quda::SolverParam::is_preconditioner, quda::TimeProfile::Last(), quda::Solver::param, quda::Solver::profile, Q, Q_AQandg, quda::QUDA_PROFILE_COMPUTE, quda::QUDA_PROFILE_EIGEN, and quda::SolverParam::secs.

Referenced by operator()().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ compute_beta()

void quda::CACG::compute_beta ( )

private

Compute the beta coefficients.

Definition at line 376 of file inv_ca_cg.cpp.

References beta, quda::SolverParam::is_preconditioner, quda::TimeProfile::Last(), quda::Solver::param, quda::Solver::profile, Q, Q_AQandg, Q_AS, quda::QUDA_PROFILE_COMPUTE, quda::QUDA_PROFILE_EIGEN, and quda::SolverParam::secs.

Referenced by operator()().

Here is the call graph for this function:

Here is the caller graph for this function:

◆ create()

void quda::CACG::create ( ColorSpinorField & b )

private

Initiate the fields needed by the solver.

Parameters

[in] b Source vector used for solver meta data. If we're not preserving the source vector and we have a uni-precision solver, we set p[0] = b to save memory and memory copying.

Definition at line 208 of file inv_ca_cg.cpp.

References alpha, AQ, AS, basis, beta, quda::Solver::constructDeflationSpace(), quda::ColorSpinorParam::create, quda::ColorSpinorField::Create(), csParam, quda::SolverParam::deflate, quda::Solver::deflate_init, quda::blas::flops, quda::CACGNR::init, quda::SolverParam::is_preconditioner, mat, quda::SolverParam::Nkrylov, quda::Solver::param, quda::SolverParam::precision, quda::SolverParam::precision_sloppy, quda::Solver::profile, Q, Q_AQandg, Q_AS, Qtmp, QUDA_NULL_FIELD_CREATE, QUDA_POWER_BASIS, quda::QUDA_PROFILE_INIT, rp, quda::ColorSpinorParam::setPrecision(), tmp_sloppy, tmp_sloppy2, tmpp, and tmpp2.