device_vec< T > Class Template Reference

#include <gpu_utils.h>

Public Member Functions
	device_vec ()
	device_vec (size_t N)
	device_vec (const device_vec< T > &src)
	device_vec (const std::vector< T > &src)
	~device_vec ()
void	init (size_t s)
void	free ()
	operator std::vector< T > () const
device_vec< T > &	operator= (const std::vector< T > &src)
device_vec< T > &	operator= (const device_vec< T > &src)
void	copyToHost (T *dst, bool async, cudaStream_t s=0)
void	copyToHost (std::vector< T > &dst, bool async, cudaStream_t s=0)
void	copyToDevice (const std::vector< T > &src, bool async=false, cudaStream_t s=0)
void	copyToDevice (const T *first, size_t size, bool async=false, cudaStream_t s=0)
std::vector< T >	toVector ()
size_t	memsize ()

Public Attributes
size_t	size
T *	data

Detailed Description

template<typename T>
class device_vec< T >

Todo:

Delete or free() not called explicitly anywhere. Needed?

Definition at line 68 of file gpu_utils.h.

Constructor & Destructor Documentation

device_vec() [1/4]

template<typename T>

device_vec< T >::device_vec ( )

inline

Definition at line 70 of file gpu_utils.h.

                 {
        data = 0;
        size = 0;
    }

device_vec() [2/4]

template<typename T>

device_vec< T >::device_vec ( size_t  N )

inline

Definition at line 75 of file gpu_utils.h.

                         { 
        data = 0;
        size = 0;
        init(N);
    }

device_vec() [3/4]

template<typename T>

device_vec< T >::device_vec ( const device_vec< T > &  src )

inline

Definition at line 80 of file gpu_utils.h.

                                         {
        data = 0; size = 0;
        init(src.size);
        dbgCUDAErrorCheck(cudaMemcpy(data, src.data, sizeof(T)*size, cudaMemcpyDeviceToDevice));
    }

device_vec() [4/4]

template<typename T>

device_vec< T >::device_vec ( const std::vector< T > &  src )

inline

Definition at line 85 of file gpu_utils.h.

                                        {
        data=0; size=0; 
        init(src.size());
        dbgCUDAErrorCheck(cudaMemcpy(data, &src[0], sizeof(T)*size, cudaMemcpyHostToDevice));
    }

~device_vec()

template<typename T>

device_vec< T >::~device_vec ( )

inline

Definition at line 90 of file gpu_utils.h.

                 {
        free();
    }

Member Function Documentation

copyToDevice() [1/2]

template<typename T>

void device_vec< T >::copyToDevice ( const std::vector< T > &  src, bool  async = false, cudaStream_t  s = 0  )

inline

Definition at line 137 of file gpu_utils.h.

                                                                                   {
        copyToDevice(&src[0], src.size(), async, s);
    }

copyToDevice() [2/2]

template<typename T>

void device_vec< T >::copyToDevice ( const T *  first, size_t  size, bool  async = false, cudaStream_t  s = 0  )

inline

Definition at line 140 of file gpu_utils.h.

                                                                                       {
        if (this->size < size)
            init(size);
        if (async)
            dbgCUDAErrorCheck(cudaMemcpyAsync(data, first, sizeof(T) * size, cudaMemcpyHostToDevice, s));
        else
            dbgCUDAErrorCheck(cudaMemcpy(data, first, sizeof(T) * size, cudaMemcpyHostToDevice));
    }

copyToHost() [1/2]

template<typename T>

void device_vec< T >::copyToHost ( T *  dst, bool  async, cudaStream_t  s = 0  )

inline

Definition at line 126 of file gpu_utils.h.

                                                          {
        if (async)
            dbgCUDAErrorCheck(cudaMemcpyAsync(dst, data, sizeof(T) * size, cudaMemcpyDeviceToHost, s));
        else
            dbgCUDAErrorCheck(cudaMemcpy(dst, data, sizeof(T) * size, cudaMemcpyDeviceToHost));
    }

copyToHost() [2/2]

template<typename T>

void device_vec< T >::copyToHost ( std::vector< T > &  dst, bool  async, cudaStream_t  s = 0  )

inline

Definition at line 132 of file gpu_utils.h.

                                                                     {
        if (dst.size() != size)
            dst.resize(size);
        copyToHost(&dst[0], async, s);
    }

free()

template<typename T>

void device_vec< T >::free ( )

inline

Definition at line 104 of file gpu_utils.h.

                {
        if (data) {
            dbgCUDAErrorCheck(cudaFree(data));
            data=0;
        }
    }

init()

template<typename T>

void device_vec< T >::init ( size_t  s )

inline

Definition at line 93 of file gpu_utils.h.

                        {
        if(size != s) {
            free();
        }
        if (s!=0) {
            if (cudaMalloc(&data, sizeof(T)*s) != cudaSuccess) {
                throw std::bad_alloc(SPrintf("device_vec<%s> init %d elements failed", typeid(T).name(), s).c_str());
            }
            size = s;
        }
    }

memsize()

template<typename T>

size_t device_vec< T >::memsize ( )

inline

Definition at line 154 of file gpu_utils.h.

{ return size*sizeof(T); }

operator std::vector< T >()

template<typename T>

device_vec< T >::operator std::vector< T > ( ) const

inline

Definition at line 110 of file gpu_utils.h.

                                  {
        std::vector<T> dst(size);
        dbgCUDAErrorCheck(cudaMemcpy(&dst[0], data, sizeof(T)*size, cudaMemcpyDeviceToHost));
        return dst;
    }

operator=() [1/2]

template<typename T>

device_vec<T>& device_vec< T >::operator= ( const std::vector< T > &  src )

inline

Definition at line 115 of file gpu_utils.h.

                                                      {
        init(src.size());
        dbgCUDAErrorCheck(cudaMemcpy(data, &src[0], sizeof(T)*size, cudaMemcpyHostToDevice));
        return *this;
    }

operator=() [2/2]

template<typename T>

device_vec<T>& device_vec< T >::operator= ( const device_vec< T > &  src )

inline

Definition at line 120 of file gpu_utils.h.

                                                       {
        clear();
        init(src.size);
        dbgCUDAErrorCheck(cudaMemcpy(data, src.data, sizeof(T)*size, cudaMemcpyDeviceToDevice));
        return *this;
    }

toVector()

template<typename T>

std::vector<T> device_vec< T >::toVector ( )

inline

Definition at line 149 of file gpu_utils.h.

                            {
        std::vector<T> v (size);
        dbgCUDAErrorCheck(cudaMemcpy(&v[0], data, sizeof(T)*size, cudaMemcpyDeviceToHost));
        return v;
    }

Member Data Documentation

data

template<typename T>

T* device_vec< T >::data

Definition at line 156 of file gpu_utils.h.

size

template<typename T>

size_t device_vec< T >::size

Definition at line 155 of file gpu_utils.h.

---

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

• cudatrack/gpu_utils.h