mirror of
https://github.com/paboyle/Grid.git
synced 2024-09-20 17:25:37 +01:00
11cb3e9a01
cut at Conjugate gradient. Also copied in Remez, Zolotarev, Chebyshev from Mike Clark, Tony Kennedy and my BFM package respectively since we know we will need these. I wanted the structure of algorithms/approx algorithms/iterative etc.. to start taking shape.
73 lines
2.2 KiB
C++
73 lines
2.2 KiB
C++
#ifndef GRID_ALGORITHM_SPARSE_MATRIX_H
|
|
#define GRID_ALGORITHM_SPARSE_MATRIX_H
|
|
|
|
#include <Grid.h>
|
|
|
|
namespace Grid {
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Interface defining what I expect of a general sparse matrix, such as a Fermion action
|
|
/////////////////////////////////////////////////////////////////////////////////////////////
|
|
template<class Field> class SparseMatrixBase {
|
|
public:
|
|
// Full checkerboar operations
|
|
virtual RealD M (const Field &in, Field &out)=0;
|
|
virtual RealD Mdag (const Field &in, Field &out)=0;
|
|
virtual void MdagM(const Field &in, Field &out,RealD &ni,RealD &no) {
|
|
Field tmp (in._grid);
|
|
ni=M(in,tmp);
|
|
no=Mdag(tmp,out);
|
|
}
|
|
};
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////
|
|
// Interface augmented by a red black sparse matrix, such as a Fermion action
|
|
/////////////////////////////////////////////////////////////////////////////////////////////
|
|
template<class Field> class CheckerBoardedSparseMatrixBase : public SparseMatrixBase<Field> {
|
|
public:
|
|
|
|
// half checkerboard operaions
|
|
virtual void Meooe (const Field &in, Field &out)=0;
|
|
virtual void Mooee (const Field &in, Field &out)=0;
|
|
virtual void MooeeInv (const Field &in, Field &out)=0;
|
|
|
|
virtual void MeooeDag (const Field &in, Field &out)=0;
|
|
virtual void MooeeDag (const Field &in, Field &out)=0;
|
|
virtual void MooeeInvDag (const Field &in, Field &out)=0;
|
|
|
|
// Schur decomp operators
|
|
virtual RealD Mpc (const Field &in, Field &out) {
|
|
Field tmp(in._grid);
|
|
|
|
Meooe(in,tmp);
|
|
MooeeInv(tmp,out);
|
|
Meooe(out,tmp);
|
|
|
|
Mooee(in,out);
|
|
out=out-tmp; // axpy_norm
|
|
RealD n=norm2(out);
|
|
return n;
|
|
}
|
|
virtual RealD MpcDag (const Field &in, Field &out){
|
|
Field tmp(in._grid);
|
|
|
|
MeooeDag(in,tmp);
|
|
MooeeInvDag(tmp,out);
|
|
MeooeDag(out,tmp);
|
|
|
|
MooeeDag(in,out);
|
|
out=out-tmp; // axpy_norm
|
|
RealD n=norm2(out);
|
|
return n;
|
|
}
|
|
virtual void MpcDagMpc(const Field &in, Field &out,RealD ni,RealD no) {
|
|
Field tmp(in._grid);
|
|
ni=Mpc(in,tmp);
|
|
no=Mpc(tmp,out);
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|