Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Protected Member Functions | Static Protected Member Functions | Protected Attributes | Static Protected Attributes
caffe2::CUDAContext Class Referencefinal

Public Member Functions

 CUDAContext (const int gpu_id=-1)
 CUDAContext (const DeviceOption &option)
void SwitchToDevice (int stream_id)
void SwitchToDevice ()
void WaitEvent (const Event &ev)
void Record (Event *ev, const char *err_msg=nullptr) const
void FinishDeviceComputation ()
int cuda_gpu_id () const
cudaStream_t cuda_stream ()
cudaStream_t cuda_stream () const
cublasHandle_t cublas_handle ()
cudnnHandle_t cudnn_handle ()
curandGenerator_t & curand_generator ()
template<class SrcContext , class DstContext >
void CopyBytes (size_t nbytes, const void *src, void *dst)
template<typename T , class SrcContext , class DstContext >
void Copy (int n, const T *src, T *dst)
template<class SrcContext , class DstContext >
void CopyItems (const TypeMeta &meta, size_t n, const void *src, void *dst)

Static Public Member Functions

static cudaStream_t cuda_stream (int gpu_id, int stream_id)
static std::pair< void *, MemoryDeleter > New (size_t nbytes)
static std::mutex & mutex ()
static std::vector< long > TotalMemoryByGpu ()
static std::vector< long > MaxMemoryByGpu ()
static bool HasAsyncPartDefault ()
static bool SupportsAsyncScheduling ()
static bool IsStreamFree (const DeviceOption &option, int stream_id)

Protected Member Functions

void set_stream_id (int stream_id)

Static Protected Member Functions

static void Delete (void *data)

Protected Attributes

int gpu_id_
int stream_id_ = 0
int random_seed_
curandGenerator_t curand_generator_ {nullptr}

Static Protected Attributes

static thread_local ThreadLocalCUDAObjects cuda_objects_

Detailed Description

Definition at line 121 of file context_gpu.h.

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