GpuMultidimArrayAtGpu< T > Class Template Reference

#include <cuda_xmipp_utils.h>

Collaboration diagram for GpuMultidimArrayAtGpu< T >:

Public Member Functions
	GpuMultidimArrayAtGpu ()
	GpuMultidimArrayAtGpu (size_t _Xdim, size_t _Ydim=1, size_t _Zdim=1, size_t _Ndim=1)
	GpuMultidimArrayAtGpu (size_t _Xdim, size_t _Ydim, size_t _Zdim, size_t _Ndim, T *deviceData)
template<typename T1 >
void	resize (const GpuMultidimArrayAtGpu< T1 > &array)
void	resize (size_t _Xdim, size_t _Ydim=1, size_t _Zdim=1, size_t _Ndim=1)
bool	isEmpty ()
void	clear ()
	~GpuMultidimArrayAtGpu ()
void	copyToGpu (T *data)
void	copyToCpu (T *data)
void	copyToGpuStream (T *data, myStreamHandle &myStream)
void	fillImageToGpu (T *data, size_t n=0)
void	fillImageToGpuStream (T *data, myStreamHandle &myStream, int n=0)
void	copyGpuToGpu (GpuMultidimArrayAtGpu< T > &gpuArray)
void	copyGpuToGpuStream (GpuMultidimArrayAtGpu< T > &gpuArray, myStreamHandle &myStream)
void	calculateGridSize (const XmippDim3 &blockSize, XmippDim3 &gridSize) const
void	calculateGridSizeVectorized (const XmippDim3 &blockSize, XmippDim3 &gridSize) const
template<typename T1 >
void	fft (GpuMultidimArrayAtGpu< T1 > &fourierTransform, mycufftHandle &myhandle)
template<typename T1 >
void	ifft (GpuMultidimArrayAtGpu< T1 > &realSpace, mycufftHandle &myhandle)
void	fftStream (GpuMultidimArrayAtGpu< std::complex< float >> &fourierTransform, mycufftHandle &myhandle, myStreamHandle &myStream, bool useCallback, GpuMultidimArrayAtGpu< std::complex< float >> &dataRef)
template<typename T1 >
void	ifftStream (GpuMultidimArrayAtGpu< T1 > &realSpace, mycufftHandle &myhandle, myStreamHandle &myStream, bool useCallback, GpuMultidimArrayAtGpu< std::complex< float > > &dataExp)
void	calculateMax (float *max_values, float *posX, float *posY, int fixPadding)

Public Attributes
size_t	Xdim
size_t	Ydim
size_t	Zdim
size_t	Ndim
size_t	yxdim
size_t	zyxdim
size_t	nzyxdim
T *	d_data

Detailed Description

template<typename T>
class GpuMultidimArrayAtGpu< T >

Definition at line 181 of file cuda_xmipp_utils.h.

Constructor & Destructor Documentation

GpuMultidimArrayAtGpu() [1/3]

template<typename T>

GpuMultidimArrayAtGpu< T >::GpuMultidimArrayAtGpu ( )

inline

Definition at line 187 of file cuda_xmipp_utils.h.

    {
        Xdim=Ydim=Zdim=Ndim=yxdim=zyxdim=nzyxdim=0;
        d_data=nullptr;
    }

GpuMultidimArrayAtGpu() [2/3]

template<typename T>

GpuMultidimArrayAtGpu< T >::GpuMultidimArrayAtGpu ( size_t  _Xdim, size_t  _Ydim = 1, size_t  _Zdim = 1, size_t  _Ndim = 1  )

inline

Definition at line 193 of file cuda_xmipp_utils.h.

    {
        Xdim=Ydim=Zdim=Ndim=yxdim=zyxdim=nzyxdim=0;
        d_data=nullptr;
        resize(_Xdim, _Ydim, _Zdim, _Ndim);
    }

GpuMultidimArrayAtGpu() [3/3]

template<typename T>

GpuMultidimArrayAtGpu< T >::GpuMultidimArrayAtGpu ( size_t  _Xdim, size_t  _Ydim, size_t  _Zdim, size_t  _Ndim, T *  deviceData  )

inline

Definition at line 200 of file cuda_xmipp_utils.h.

    {
        setDims(_Xdim, _Ydim, _Zdim, _Ndim);
        d_data = deviceData;
    }

~GpuMultidimArrayAtGpu()

template<typename T>

GpuMultidimArrayAtGpu< T >::~GpuMultidimArrayAtGpu ( )

inline

Definition at line 230 of file cuda_xmipp_utils.h.

    {
        clear();
    }

Member Function Documentation

calculateGridSize()

template<typename T>

void GpuMultidimArrayAtGpu< T >::calculateGridSize ( const XmippDim3 &  blockSize, XmippDim3 &  gridSize  ) const

inline

Definition at line 276 of file cuda_xmipp_utils.h.

    {
        gridSize.x=gridFromBlock(Xdim,blockSize.x);
        gridSize.y=gridFromBlock(Ydim,blockSize.y);
        gridSize.z=gridFromBlock(Zdim,blockSize.z);
    }

calculateGridSizeVectorized()

template<typename T>

void GpuMultidimArrayAtGpu< T >::calculateGridSizeVectorized ( const XmippDim3 &  blockSize, XmippDim3 &  gridSize  ) const

inline

Definition at line 283 of file cuda_xmipp_utils.h.

    {
        gridSize.x=gridFromBlock(nzyxdim,blockSize.x);
        gridSize.y=1;
        gridSize.z=1;
    }

calculateMax()

template<typename T>

void GpuMultidimArrayAtGpu< T >::calculateMax	(	float *	max_values,
		float *	posX,
		float *	posY,
		int	fixPadding
	)

clear()

template<typename T>

void GpuMultidimArrayAtGpu< T >::clear ( )

inline

Definition at line 220 of file cuda_xmipp_utils.h.

    {
        if (d_data!=nullptr){
            gpuFree((void*) d_data);

        }
        Xdim=Ydim=Zdim=Ndim=yxdim=zyxdim=nzyxdim=0;
        d_data=nullptr;
    }

copyGpuToGpu()

template<typename T>

void GpuMultidimArrayAtGpu< T >::copyGpuToGpu ( GpuMultidimArrayAtGpu< T > &  gpuArray )

inline

Definition at line 260 of file cuda_xmipp_utils.h.

    {
        if (gpuArray.isEmpty())
            gpuArray.resize(Xdim,Ydim,Zdim,Ndim);

        gpuCopyFromGPUToGPU(d_data, gpuArray.d_data, nzyxdim*sizeof(T));
    }

copyGpuToGpuStream()

template<typename T>

void GpuMultidimArrayAtGpu< T >::copyGpuToGpuStream ( GpuMultidimArrayAtGpu< T > &  gpuArray, myStreamHandle &  myStream  )

inline

Definition at line 268 of file cuda_xmipp_utils.h.

    {
        if (gpuArray.isEmpty())
            gpuArray.resize(Xdim,Ydim,Zdim,Ndim);

        gpuCopyFromGPUToGPUStream(d_data, gpuArray.d_data, nzyxdim*sizeof(T), myStream);
    }

copyToCpu()

template<typename T>

void GpuMultidimArrayAtGpu< T >::copyToCpu ( T *  data )

inline

Definition at line 240 of file cuda_xmipp_utils.h.

    {
        gpuCopyFromGPUToCPU((void *)d_data, (void *)data, nzyxdim*sizeof(T));
    }

copyToGpu()

template<typename T>

void GpuMultidimArrayAtGpu< T >::copyToGpu ( T *  data )

inline

Definition at line 235 of file cuda_xmipp_utils.h.

    {
        gpuCopyFromCPUToGPU((void *)data, (void *)d_data, nzyxdim*sizeof(T));
    }

copyToGpuStream()

template<typename T>

void GpuMultidimArrayAtGpu< T >::copyToGpuStream ( T *  data, myStreamHandle &  myStream  )

inline

Definition at line 245 of file cuda_xmipp_utils.h.

    {
        gpuCopyFromCPUToGPUStream((void *)data, (void *)d_data, nzyxdim*sizeof(T), myStream);
    }

fft()

template<typename T>

template<typename T1 >

void GpuMultidimArrayAtGpu< T >::fft	(	GpuMultidimArrayAtGpu< T1 > &	fourierTransform,
		mycufftHandle &	myhandle
	)

fftStream()

template<typename T>

void GpuMultidimArrayAtGpu< T >::fftStream	(	GpuMultidimArrayAtGpu< std::complex< float >> &	fourierTransform,
		mycufftHandle &	myhandle,
		myStreamHandle &	myStream,
		bool	useCallback,
		GpuMultidimArrayAtGpu< std::complex< float >> &	dataRef
	)

fillImageToGpu()

template<typename T>

void GpuMultidimArrayAtGpu< T >::fillImageToGpu ( T *  data, size_t  n = 0  )

inline

Definition at line 250 of file cuda_xmipp_utils.h.

    {
        gpuCopyFromCPUToGPU((void *)data, (void *)&d_data[n*zyxdim], zyxdim*sizeof(T));
    }

fillImageToGpuStream()

template<typename T>

void GpuMultidimArrayAtGpu< T >::fillImageToGpuStream ( T *  data, myStreamHandle &  myStream, int  n = 0  )

inline

Definition at line 255 of file cuda_xmipp_utils.h.

    {
        gpuCopyFromCPUToGPUStream((void *)data, (void *)&d_data[n*zyxdim], zyxdim*sizeof(T), myStream);
    }

ifft()

template<typename T>

template<typename T1 >

void GpuMultidimArrayAtGpu< T >::ifft	(	GpuMultidimArrayAtGpu< T1 > &	realSpace,
		mycufftHandle &	myhandle
	)

ifftStream()

template<typename T>

template<typename T1 >

void GpuMultidimArrayAtGpu< T >::ifftStream	(	GpuMultidimArrayAtGpu< T1 > &	realSpace,
		mycufftHandle &	myhandle,
		myStreamHandle &	myStream,
		bool	useCallback,
		GpuMultidimArrayAtGpu< std::complex< float > > &	dataExp
	)

isEmpty()

template<typename T>

bool GpuMultidimArrayAtGpu< T >::isEmpty ( )

inline

Definition at line 215 of file cuda_xmipp_utils.h.

    {
        return d_data==nullptr;
    }

resize() [1/2]

template<typename T>

template<typename T1 >

void GpuMultidimArrayAtGpu< T >::resize ( const GpuMultidimArrayAtGpu< T1 > &  array )

inline

Definition at line 207 of file cuda_xmipp_utils.h.

    {

        resize(array.Xdim, array.Ydim, array.Zdim, array.Ndim);
    }

resize() [2/2]

template<typename T >

void GpuMultidimArrayAtGpu< T >::resize	(	size_t	_Xdim,
		size_t	_Ydim = 1,
		size_t	_Zdim = 1,
		size_t	_Ndim = 1
	)

Definition at line 44 of file cuda_xmipp_utils.cpp.

{
    if (_Xdim*_Ydim*_Zdim*_Ndim==nzyxdim){

        return;
    }

    clear();

    setDims(_Xdim, _Ydim, _Zdim, _Ndim);
    gpuErrchk(cudaMalloc(&d_data,nzyxdim*sizeof(T)));
}

Member Data Documentation

d_data

template<typename T>

T* GpuMultidimArrayAtGpu< T >::d_data

Definition at line 185 of file cuda_xmipp_utils.h.

Ndim

template<typename T>

size_t GpuMultidimArrayAtGpu< T >::Ndim

Definition at line 184 of file cuda_xmipp_utils.h.

nzyxdim

template<typename T>

size_t GpuMultidimArrayAtGpu< T >::nzyxdim

Definition at line 184 of file cuda_xmipp_utils.h.

Xdim

template<typename T>

size_t GpuMultidimArrayAtGpu< T >::Xdim

Definition at line 184 of file cuda_xmipp_utils.h.

Ydim

template<typename T>

size_t GpuMultidimArrayAtGpu< T >::Ydim

Definition at line 184 of file cuda_xmipp_utils.h.

yxdim

template<typename T>

size_t GpuMultidimArrayAtGpu< T >::yxdim

Definition at line 184 of file cuda_xmipp_utils.h.

Zdim

template<typename T>

size_t GpuMultidimArrayAtGpu< T >::Zdim

Definition at line 184 of file cuda_xmipp_utils.h.

zyxdim

template<typename T>

size_t GpuMultidimArrayAtGpu< T >::zyxdim

Definition at line 184 of file cuda_xmipp_utils.h.

---

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

• xmipp/libraries/reconstruction_cuda/cuda_xmipp_utils.h
• xmipp/libraries/reconstruction_cuda/cuda_xmipp_utils.cpp