|
|
| 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 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) |
| |
|
|
void | set_stream_id (int stream_id) |
| |
|
|
static void | Delete (void *data) |
| |
|
|
int | gpu_id_ |
| |
|
int | stream_id_ = 0 |
| |
|
int | random_seed_ |
| |
|
curandGenerator_t | curand_generator_ {nullptr} |
| |
Definition at line 121 of file context_gpu.h.
The documentation for this class was generated from the following file:
1.8.11
