Gromacs
2025-dev-20240906-b7d3419
|
Impl class.
Public Member Functions | |
Impl (bool allocateRealGrid, MPI_Comm comm, ArrayRef< const int > gridSizesInXForEachRank, ArrayRef< const int > gridSizesInYForEachRank, const int nz, bool performOutOfPlaceFFT, const DeviceContext &context, const DeviceStream &pmeStream, ivec realGridSize, ivec realGridSizePadded, ivec complexGridSizePadded, DeviceBuffer< float > *realGrid, DeviceBuffer< float > *complexGrid) | |
Default constructor. More... | |
Public Attributes | |
RocfftInitializer | init_ |
Handle initializing the rocFFT library. More... | |
EnumerationArray< FftDirection, RocfftPlan > | plans_ |
Data for 3D FFT plans and execution. | |
float * | realGrid_ |
Handle to the real grid buffer. | |
float * | complexGrid_ |
sycl::queue | queue_ |
Copy of PME stream. More... | |
gmx::Gpu3dFft::ImplSyclRocfft::Impl::Impl | ( | bool | allocateRealGrid, |
MPI_Comm | comm, | ||
ArrayRef< const int > | gridSizesInXForEachRank, | ||
ArrayRef< const int > | gridSizesInYForEachRank, | ||
const int | nz, | ||
bool | performOutOfPlaceFFT, | ||
const DeviceContext & | context, | ||
const DeviceStream & | pmeStream, | ||
ivec | realGridSize, | ||
ivec | realGridSizePadded, | ||
ivec | complexGridSizePadded, | ||
DeviceBuffer< float > * | realGrid, | ||
DeviceBuffer< float > * | complexGrid | ||
) |
Default constructor.
RocfftInitializer gmx::Gpu3dFft::ImplSyclRocfft::Impl::init_ |
Handle initializing the rocFFT library.
Make sure the library is initialized before the plans, etc. and not destructed before they are.
sycl::queue gmx::Gpu3dFft::ImplSyclRocfft::Impl::queue_ |
Copy of PME stream.
This copy is guaranteed by the SYCL standard to work as if it was the original.