Skip to content

Class TempLat::KokkosFFTMemoryLayout

template <size_t NDim>

ClassList > TempLat > KokkosFFTMemoryLayout

Implements part of FFTLibraryInterface , computes the local memory associated to the global problem, for KokkosFFT. Build on this to replace KokkosFFT.More...

  • #include <kokkosfftmemorylayout.h>

Inherits the following classes: TempLat::KokkosFFTPlanner

Public Functions

Type Name
KokkosFFTMemoryLayout () = default
virtual FFTLayoutStruct< NDim > computeLocalSizes (MPICartesianGroup group, device::IdxArray< NDim > nGridPoints, bool forbidTransposition=!KOKKOSFFTWITHTRANSPOSITION())
given an actual setup, return the description of the subarray of the global problem that this process holds.

Public Functions inherited from TempLat::KokkosFFTPlanner

See TempLat::KokkosFFTPlanner

Type Name
KokkosFFTPlanner () = default
virtual std::shared_ptr< FFTPlanInterface< double, NDim > > getPlans_double (const MPICartesianGroup & group, const FFTLayoutStruct< NDim > & layout)
Create fully working plans, which must self-destruct in the FFTPlanInterface 's destructor. Use shared_ptr's.
virtual std::shared_ptr< FFTPlanInterface< float, NDim > > getPlans_float (const MPICartesianGroup & group, const FFTLayoutStruct< NDim > & layout)
Create fully working plans, which must self-destruct in the FFTPlanInterface 's destructor. Use shared_ptr's.
virtual void setPlannerPatience (int level)
If your library has different levels of patience for the planning phase, set it here.

Public Functions inherited from TempLat::FFTLibraryInterface

See TempLat::FFTLibraryInterface

Type Name
FFTLibraryInterface () = default
virtual FFTLayoutStruct< NDim > computeLocalSizes (MPICartesianGroup group, device::IdxArray< NDim > nGridPoints, bool forbidTransposition=false) = 0
given an actual setup, return the description of the subarray of the global problem that this process holds.
virtual IntrinsicScales getIntrinsicRescaleToGetUnnormalizedFFT (device::Idx nGridPoints) = 0
The lattice objects expect an unnormalized FFT, such as the one FFTW gives: applying once forward and then backward should return the input values multiplied by nGridPoints^nDimensions, i.e. the total number of values in the problem. If your library does not respect that, return the factor by which we must multiply the output values to obtain the unnormalized FFT. Return defaultIntrinsicScales (exact (double) 1) in order not to waste time, if your library does it like FFTW.
virtual device::Idx getMaximumNumberOfDimensionsToDivide (device::Idx nDimensions) = 0
Return the number of dimensions over which the MPI processes can be distributed, given a problem of dimension nDimensions. FFTW can only divide in 1 dimension: the leading dimension. ParaFaFT can divide an N dimensional setup over the leading N - 1 dimensions.
virtual std::shared_ptr< FFTPlanInterface< double, NDim > > getPlans_double (const MPICartesianGroup & group, const FFTLayoutStruct< NDim > & layout) = 0
Create fully working plans, which must self-destruct in the FFTPlanInterface 's destructor. Use shared_ptr's.
virtual std::shared_ptr< FFTPlanInterface< float, NDim > > getPlans_float (const MPICartesianGroup & group, const FFTLayoutStruct< NDim > & layout) = 0
Create fully working plans, which must self-destruct in the FFTPlanInterface 's destructor. Use shared_ptr's. Since we use virtual methods here, we cannot use templates. Only one type of dynamic typing allowed by C++, either runtime (virtual) or compile time (template).
virtual void setPlannerPatience (int level) = 0
If your library has different levels of patience for the planning phase, set it here.
virtual ~FFTLibraryInterface () = default

Public Static Functions

Type Name
constexpr bool KOKKOSFFTWITHTRANSPOSITION ()

Detailed Description

Unit test: ctest -R test-kokkosfftmemorylayout

Public Functions Documentation

function KokkosFFTMemoryLayout

TempLat::KokkosFFTMemoryLayout::KokkosFFTMemoryLayout () = default

function computeLocalSizes

given an actual setup, return the description of the subarray of the global problem that this process holds.

inline virtual FFTLayoutStruct < NDim > TempLat::KokkosFFTMemoryLayout::computeLocalSizes (
    MPICartesianGroup group,
    device::IdxArray< NDim > nGridPoints,
    bool forbidTransposition=!KOKKOSFFTWITHTRANSPOSITION()
) 

Implements TempLat::FFTLibraryInterface::computeLocalSizes


Public Static Functions Documentation

function KOKKOSFFTWITHTRANSPOSITION

static inline constexpr bool TempLat::KokkosFFTMemoryLayout::KOKKOSFFTWITHTRANSPOSITION () 


The documentation for this class was generated from the following file cosmolatticeweb/tmp/code_source/templat/include/TempLat/fft/external/kokkosfft/kokkosfftmemorylayout.h