|
Mila 0.13.48
Deep Neural Network Library
|
DeviceTypeTraits specialization for CUDA devices. More...
#include <type_traits>import Compute.CudaDeviceMemoryResource;import Compute.DeviceType;import Compute.DeviceTypeTraits;Classes | |
| struct | Mila::Dnn::Compute::DeviceTypeTraits< DeviceType::Cuda > |
| DeviceTypeTraits specialization for the CUDA device. More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
DeviceTypeTraits specialization for CUDA devices.
Provides the canonical mapping from Mila::Dnn::Compute::DeviceType::Cuda to the preferred memory resource type used for CUDA-backed tensors.