Collaboration diagram for gmx::Gpu3dFft::ImplHipRocfft::Impl:

Description

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_

const DeviceStream &	pmeStream_

Constructor & Destructor Documentation

gmx::Gpu3dFft::ImplHipRocfft::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::ImplHipRocfft::Impl::init_

Handle initializing the rocFFT library.

Make sure the library is initialized before the plans, etc. and not destructed before they are.

The documentation for this class was generated from the following file: