|
Mila 0.13.48
Deep Neural Network Library
|
Implementation of CUDA-based compute device for the Mila framework. More...
#include <cuda_runtime.h>#include <string>#include <format>#include <stdexcept>#include <utility>#include <memory>#include <exception>import Logging.Logger;import Cuda.Error;import Compute.DeviceRegistry;import Compute.CudaDeviceProps;import Compute.DeviceType;import Compute.DeviceId;import Compute.Device;Classes | |
| class | Mila::Dnn::Compute::CudaDevice |
| Class representing a CUDA compute device instance. More... | |
| class | Mila::Dnn::Compute::CudaDeviceRegistrar |
| CUDA device registrar for device-agnostic registration. More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
Implementation of CUDA-based compute device for the Mila framework.