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
| 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
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
The documentation for this class was generated from the following file cosmolatticeweb/tmp/code_source/templat/include/TempLat/fft/external/kokkosfft/kokkosfftmemorylayout.h