dragon.cuda

Classes

class Stream : The CUDA stream wrapper.

Functions

current_device(…) : Return the index of current selected device.

enable_cudnn(…) : Enable backend to use the cuDNN library.

get_device_capability(…) : Return the capability of specified device.

is_available(…) : Return a bool reporting if runtime is available.

memory_allocated(…) : Return the size of memory used by tensors in current workspace.

set_default_device(…) : Set the default device.

set_device(…) : Set the current device.

synchronize(…) : Synchronize a specified CUDA stream.