cuda_xmipp_utils.h File Reference

#include <stdio.h>
#include <complex>

Include dependency graph for cuda_xmipp_utils.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	mycufftHandle
class	myStreamHandle
class	XmippDim3
class	TransformMatrix< T >
class	GpuMultidimArrayAtGpu< T >

Macros
#define	CONVERT2DIM3(d)   (dim3((d).x,(d).y,(d).z))

Functions
void	mycufftDestroy (void *ptr)
void	myStreamDestroy (void *ptr)
void	myStreamCreate (myStreamHandle &myStream)
void	gpuMalloc (void **d_data, size_t Nbytes)
void	gpuFree (void *d_data)
void	cpuMalloc (void **h_data, size_t Nbytes)
void	cpuFree (void *h_data)
void	initializeIdentity (float *d_data, float *h_data, int Ndim, myStreamHandle &myStream)
void	setTranslationMatrix (float *d_data, float *posX, float *posY, int Ndim, myStreamHandle &myStream)
void	setRotationMatrix (float *d_data, float *ang, int Ndim, myStreamHandle &myStream)
void	gpuCopyFromGPUToGPUStream (void *d_dataFrom, void *d_dataTo, size_t Nbytes, myStreamHandle &myStream)
void	gpuCopyFromCPUToGPUStream (void *data, void *d_data, size_t Nbytes, myStreamHandle &myStream)
void	gpuCopyFromGPUToCPUStream (void *d_data, void *data, size_t Nbytes, myStreamHandle &myStream)
void	gpuCopyFromGPUToGPU (void *d_dataFrom, void *d_dataTo, size_t Nbytes)
void	gpuCopyFromCPUToGPU (void *data, void *d_data, size_t Nbytes)
void	gpuCopyFromGPUToCPU (void *d_data, void *data, size_t Nbytes)
int	gridFromBlock (int tasks, int Nthreads)
template<typename T >
T *	loadToGPU (const T *data, size_t items)
void	cuda_check_gpu_memory (float *data)
void	cuda_check_gpu_properties (int *maxGridSize)

Function Documentation

cpuFree()

void cpuFree

(

void *

h_data

)

Definition at line 211 of file cuda_xmipp_utils.cpp.

{
    gpuErrchk(cudaFreeHost(h_data));
}

cpuMalloc()

void cpuMalloc	(	void **	h_data,
		size_t	Nbytes
	)

Definition at line 206 of file cuda_xmipp_utils.cpp.

{
    gpuErrchk(cudaMallocHost(h_data, Nbytes));
}

gpuCopyFromCPUToGPU()

void gpuCopyFromCPUToGPU	(	void *	data,
		void *	d_data,
		size_t	Nbytes
	)

Definition at line 278 of file cuda_xmipp_utils.cpp.

{
    gpuErrchk(cudaMemcpy(d_data, data, Nbytes, cudaMemcpyHostToDevice));
}

gpuCopyFromCPUToGPUStream()

void gpuCopyFromCPUToGPUStream	(	void *	data,
		void *	d_data,
		size_t	Nbytes,
		myStreamHandle &	myStream
	)

Definition at line 293 of file cuda_xmipp_utils.cpp.

{
    cudaStream_t *stream = (cudaStream_t*) myStream.ptr;
    gpuErrchk(cudaMemcpyAsync(d_data, data, Nbytes, cudaMemcpyHostToDevice, *stream));

    //gpuErrchk(cudaStreamSynchronize(*stream));
}

gpuCopyFromGPUToCPU()

void gpuCopyFromGPUToCPU	(	void *	d_data,
		void *	data,
		size_t	Nbytes
	)

Definition at line 283 of file cuda_xmipp_utils.cpp.

{
    gpuErrchk(cudaMemcpy(data, d_data, Nbytes, cudaMemcpyDeviceToHost));
}

gpuCopyFromGPUToCPUStream()

void gpuCopyFromGPUToCPUStream	(	void *	d_data,
		void *	data,
		size_t	Nbytes,
		myStreamHandle &	myStream
	)

Definition at line 301 of file cuda_xmipp_utils.cpp.

{
    cudaStream_t *stream = (cudaStream_t*) myStream.ptr;
    gpuErrchk(cudaMemcpyAsync(data, d_data, Nbytes, cudaMemcpyDeviceToHost, *stream));

    gpuErrchk(cudaStreamSynchronize(*stream));
    //cudaDeviceSynchronize();
}

gpuCopyFromGPUToGPU()

void gpuCopyFromGPUToGPU	(	void *	d_dataFrom,
		void *	d_dataTo,
		size_t	Nbytes
	)

Definition at line 288 of file cuda_xmipp_utils.cpp.

{
    gpuErrchk(cudaMemcpy(d_dataTo, d_dataFrom, Nbytes, cudaMemcpyDeviceToDevice));
}

gpuCopyFromGPUToGPUStream()

void gpuCopyFromGPUToGPUStream	(	void *	d_dataFrom,
		void *	d_dataTo,
		size_t	Nbytes,
		myStreamHandle &	myStream
	)

Definition at line 310 of file cuda_xmipp_utils.cpp.

{
    cudaStream_t *stream = (cudaStream_t*) myStream.ptr;
    gpuErrchk(cudaMemcpyAsync(d_dataTo, d_dataFrom, Nbytes, cudaMemcpyDeviceToDevice, *stream));
}

gpuFree()

void gpuFree

(

void *

d_data

)

Definition at line 201 of file cuda_xmipp_utils.cpp.

{
    gpuErrchk(cudaFree(d_data));
}

gpuMalloc()

void gpuMalloc	(	void **	d_data,
		size_t	Nbytes
	)

Definition at line 196 of file cuda_xmipp_utils.cpp.

{
    gpuErrchk(cudaMalloc(d_data, Nbytes));
}

gridFromBlock()

int gridFromBlock	(	int	tasks,
		int	Nthreads
	)

Definition at line 316 of file cuda_xmipp_utils.cpp.

{
    int numBlk = tasks/Nthreads;
    if(tasks%Nthreads>0)
        numBlk++;
    return numBlk;
}

initializeIdentity()

void initializeIdentity	(	float *	d_data,
		float *	h_data,
		int	Ndim,
		myStreamHandle &	myStream
	)

Definition at line 216 of file cuda_xmipp_utils.cpp.

{
    //float *matrices = new float[Ndim*9];
    for(int i=0; i<Ndim; i++){
        float aux_matrix[9] = {1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0};
        int offset=i*9;
        for (int j=0; j<9; j++)
            h_data[offset+j] = aux_matrix[j];
    }
    cudaStream_t *stream = (cudaStream_t*) myStream.ptr;
    gpuErrchk(cudaMemcpyAsync((void*)d_data, h_data, Ndim*9*sizeof(float), cudaMemcpyHostToDevice, *stream));
    //delete []matrices;

}

mycufftDestroy()

void mycufftDestroy

(

void *

ptr

)

Definition at line 72 of file cuda_xmipp_utils.cpp.

{
    cufftHandle *planPtr = (cufftHandle *)ptr;
    cufftDestroy(*planPtr);
    delete planPtr;
}

myStreamCreate()

void myStreamCreate

(

myStreamHandle &

myStream

)

Definition at line 63 of file cuda_xmipp_utils.cpp.

{
    cudaStream_t *streamPtr = new cudaStream_t;
    gpuErrchk(cudaStreamCreate(streamPtr));
    myStream.ptr = (void*)streamPtr;
    //printf("ptr %p\n", myStream.ptr);
    //printf("streamPtr %p\n", streamPtr);
}

myStreamDestroy()

void myStreamDestroy

(

void *

ptr

)

Definition at line 57 of file cuda_xmipp_utils.cpp.

{
    cudaStream_t *streamPtr = (cudaStream_t *)ptr;
    cudaStreamDestroy(*streamPtr);
}

setRotationMatrix()

void setRotationMatrix	(	float *	d_data,
		float *	ang,
		int	Ndim,
		myStreamHandle &	myStream
	)

Definition at line 260 of file cuda_xmipp_utils.cpp.

{

    float *rad_vector;
    gpuErrchk(cudaMallocHost((void**)&rad_vector, sizeof(float)*Ndim*9));

    for(int i=0; i<Ndim; i++){
        float rad = (float)(-ang[i]*M_PI/180);
        float matrix[9] = {cosf(rad), -sinf(rad), 0, sinf(rad), cosf(rad), 0, 0, 0, 1};
        int offset=i*9;
        for (int j=0; j<9; j++)
            rad_vector[offset+j] = matrix[j];
    }
    cudaStream_t *stream = (cudaStream_t*) myStream.ptr;
    gpuErrchk(cudaMemcpyAsync((void*)d_data, rad_vector, Ndim*9*sizeof(float), cudaMemcpyHostToDevice, *stream));
    delete []rad_vector;
}

setTranslationMatrix()

void setTranslationMatrix	(	float *	d_data,
		float *	posX,
		float *	posY,
		int	Ndim,
		myStreamHandle &	myStream
	)

Definition at line 237 of file cuda_xmipp_utils.cpp.

{
    float *matrices;
    gpuErrchk(cudaMallocHost((void**)&matrices, sizeof(float)*Ndim*9));

    for(int i=0; i<Ndim; i++){
        float aux_matrix[9] = {1, 0, -posX[i], 0, 1, -posY[i], 0, 0, 1};
        int offset=i*9;
        //memcpy(&matrices[offset], &aux_matrix, 9*sizeof(float));
        for (int j=0; j<9; j++)
            matrices[offset+j] = aux_matrix[j];
    }
    cudaStream_t *stream = (cudaStream_t*) myStream.ptr;
    gpuErrchk(cudaMemcpyAsync((void*)d_data, matrices, Ndim*9*sizeof(float), cudaMemcpyHostToDevice, *stream));
    delete []matrices;
}