Communication-avoiding GCR solver. This solver does un-preconditioned GCR, first building up a polynomial in the linear operator of length nKrylov, and then performs a minimum residual extrapolation on the resulting basis vectors. For use as a multigrid smoother with minimum global synchronization. More...

#include <invert_quda.h>

Inheritance diagram for quda::CAGCR:

[legend]

Collaboration diagram for quda::CAGCR:

[legend]

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

virtual	~CAGCR ()

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	solve (Complex psi_, std::vector< ColorSpinorField > &q, ColorSpinorField &b)
	Solve the equation A p_k psi_k = q_k psi_k = b by minimizing the least square residual using Eigen's LDLT Cholesky for numerical stability. More...

Private Attributes
const DiracMatrix &	mat

const DiracMatrix &	matSloppy

bool	init

QudaCABasis	basis

Complex *	alpha

ColorSpinorField *	rp

ColorSpinorField *	tmpp

ColorSpinorField *	tmp_sloppy

std::vector< ColorSpinorField * >	p

std::vector< ColorSpinorField * >	q

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 GCR solver. This solver does un-preconditioned GCR, first building up a polynomial in the linear operator of length nKrylov, and then performs a minimum residual extrapolation on the resulting basis vectors. For use as a multigrid smoother with minimum global synchronization.

Definition at line 990 of file invert_quda.h.

Constructor & Destructor Documentation

◆ CAGCR()

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

Definition at line 7 of file inv_ca_gcr.cpp.

◆ ~CAGCR()

quda::CAGCR::~CAGCR ( )

virtual

Member Function Documentation

◆ create()

void quda::CAGCR::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 40 of file inv_ca_gcr.cpp.

References alpha, basis, quda::Solver::constructDeflationSpace(), quda::ColorSpinorParam::create, quda::ColorSpinorField::Create(), csParam, quda::SolverParam::deflate, quda::Solver::deflate_init, quda::DiracMatrix::Expose(), quda::blas::flops, init, quda::SolverParam::is_preconditioner, mat, quda::SolverParam::Nkrylov, p, quda::Solver::param, quda::SolverParam::precision, quda::SolverParam::precision_sloppy, quda::SolverParam::preserve_source, quda::Solver::profile, q, QUDA_NULL_FIELD_CREATE, QUDA_POWER_BASIS, QUDA_PRESERVE_SOURCE_NO, quda::QUDA_PROFILE_INIT, QUDA_USE_INIT_GUESS_NO, rp, quda::ColorSpinorParam::setPrecision(), tmp_sloppy, tmpp, quda::SolverParam::use_init_guess, and warningQuda.