Gromacs
2025-dev-20240906-b7d3419
|
Builds a kernel to convert between PME and FFT grids.
pmeToFft | A boolean which tells if this is conversion from PME grid to FFT grid or reverse |
Static Public Member Functions | |
template<int subGroupSize> | |
static auto | convertKernel (sycl::uint3 fftNData, float *__restrict__ d_realGrid, float *__restrict__ d_fftGrid, sycl::uint3 fftSize, sycl::uint3 pmeSize) |
Returns the conversion kernel. More... | |
template<int subGroupSize, class... Args> | |
static void | submit (const DeviceStream &deviceStream, sycl::uint3 localFftNData, Args &&...args) |
Submits kernel. More... | |
|
inlinestatic |
Returns the conversion kernel.
[in] | fftNData | Local FFT grid size without padding |
[in,out] | d_realGrid | Local PME grid |
[in,out] | d_fftGrid | Local FFT grid |
[in] | fftSize | Local FFT grid padded size |
[in] | pmeSize | Local PME grid padded size |
subGroupSize | Size of the sub-group. |
|
inlinestatic |
Submits kernel.
[in] | deviceStream | The device stream upon which to submit |
[in] | localFftNData | Local FFT grid size without padding |
[in] | args | Parameter pack to pass to the kernel builder |
subGroupSize | Size of the sub-group. |