portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2026-09-04 16:59:36 +01:00
Code Releases Activity

Use PlannedFFT in momentum space propagator.

Browse Source
This commit is contained in:
Peter Boyle
2026-09-03 20:59:42 -04:00
parent 59f4a3729a
commit 953a401366
5 changed files with 26 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Grid/qcd/action/fermion/ContinuedFractionFermion5D.h
+3 -2
View File
@@ -65,8 +65,9 @@ public:
std::cout << "Free Propagator for PartialFraction"<<std::endl;
FermionField in_k(in.Grid());
FermionField prop_k(in.Grid());
FFT theFFT((GridCartesian *) in.Grid());
GRID_ASSERT(in.Grid() == this->FermionGrid());
PlannedFFT<typename FermionField::vector_object> &theFFT = this->FermionGridFFT();
//phase for boundary condition
ComplexField coor(in.Grid());
Grid/qcd/action/fermion/DomainWallFermion.h
+2 -1
View File
@@ -45,7 +45,8 @@ public:
FermionField in_k(in.Grid());
FermionField prop_k(in.Grid());
FFT theFFT((GridCartesian *) in.Grid());
GRID_ASSERT(in.Grid() == this->FermionGrid());
PlannedFFT<typename FermionField::vector_object> &theFFT = this->FermionGridFFT();
//phase for boundary condition
ComplexField coor(in.Grid());
Grid/qcd/action/fermion/FermionOperator.h
+17 -3
View File
@@ -45,7 +45,7 @@ public:
INHERIT_IMPL_TYPES(Impl);
FermionOperator(const ImplParams &p= ImplParams()) : Impl(p) {};
virtual ~FermionOperator(void) = default;
virtual ~FermionOperator(void) { if ( _fermionGridFFT ) delete _fermionGridFFT; }
virtual FermionField &tmp(void) = 0;
@@ -95,9 +95,23 @@ public:
virtual void MomentumSpacePropagator(FermionField &out,const FermionField &in,RealD _m,std::vector<double> twist) { GRID_ASSERT(0);};
virtual void FreePropagator(const FermionField &in,FermionField &out,RealD mass,std::vector<Complex> boundary,std::vector<double> twist)
protected:
// Cached planned FFT on the fermion grid -- a general utility (FreePropagator
// today; smoother, smearings and other users anticipated). Frontier FFTW plan
// create+destroy is ~22 ms/call (measured, Test_fft_prop PLANCOST) -- ~4x the
// transform itself and ~80% of an unplanned call -- so a per-call `FFT theFFT(grid)`
// dominates. Lazily built once on FermionGrid() and reused; owned, deleted in the dtor.
PlannedFFT<typename FermionField::vector_object> *_fermionGridFFT{nullptr};
PlannedFFT<typename FermionField::vector_object> & FermionGridFFT(void) {
if ( _fermionGridFFT == nullptr )
_fermionGridFFT = new PlannedFFT<typename FermionField::vector_object>((GridCartesian *)this->FermionGrid());
return *_fermionGridFFT;
}
public:
virtual void FreePropagator(const FermionField &in,FermionField &out,RealD mass,std::vector<Complex> boundary,std::vector<double> twist)
{
FFT theFFT((GridCartesian *) in.Grid());
GRID_ASSERT(in.Grid() == this->FermionGrid());
PlannedFFT<typename FermionField::vector_object> &theFFT = this->FermionGridFFT();
typedef typename Simd::scalar_type Scalar;
Grid/qcd/action/fermion/PartialFractionFermion5D.h
+3 -2
View File
@@ -95,8 +95,9 @@ public:
std::cout << "Free Propagator for PartialFraction"<<std::endl;
FermionField in_k(in.Grid());
FermionField prop_k(in.Grid());
FFT theFFT((GridCartesian *) in.Grid());
GRID_ASSERT(in.Grid() == this->FermionGrid());
PlannedFFT<typename FermionField::vector_object> &theFFT = this->FermionGridFFT();
//phase for boundary condition
ComplexField coor(in.Grid());
tests/core/Test_fft_prop.cc
+1 -1
View File
@@ -46,7 +46,7 @@ void bench(GridCartesian *grid, std::string name)
LatticeObject S(grid); gaussian(RNG,S);
typedef typename LatticeObject::vector_object vobj;
const int nrep = 10;
const int nrep = 50;
// Correctness + Parseval pass (unplanned), also the WARMUP that absorbs
// FFTW/allocator/MPI first-touch so the two timed passes below are both warm