dragon.cuda

Functions

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

enable_cudnn(…) : Activate the CuDNN engine.

enable_cudnn_benchmark(…) : Activate the CuDNN benchmark.

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

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

set_default_device(…) : Set the default device.

set_device(…) : Set the current device.

synchronize(…) : Synchronize the specified cuda stream.