dragon.cuda¶

Classes¶

class Stream : The CUDA stream wrapper.

Functions¶

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

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

get_device_name(…) : Return the name 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_cublas_flags(…) : Set the flags of cuBLAS library.

set_cudnn_flags(…) : Set the flags of cuDNN library.

set_default_device(…) : Set the default device.

set_device(…) : Set the current device.

synchronize(…) : Synchronize a specified CUDA stream.