|
Mila 0.13.48
Deep Neural Network Library
|
Device type definitions and conversion utilities for compute devices. More...
#include <stdexcept>#include <algorithm>#include <string>#include <cctype>#include <string_view>#include <format>Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
Enumerations | |
| enum class | Mila::Dnn::Compute::DeviceType { Mila::Dnn::Compute::Cpu , Mila::Dnn::Compute::Cuda , Mila::Dnn::Compute::Metal , Mila::Dnn::Compute::Rocm } |
| Enumeration of supported compute device types. More... | |
Functions | |
| std::string | Mila::Dnn::Compute::deviceTypeToString (DeviceType device_type) |
| Converts a DeviceType to its string representation. | |
| DeviceType | Mila::Dnn::Compute::toDeviceType (std::string_view device_type) |
| Converts a string to the corresponding DeviceType. | |
Device type definitions and conversion utilities for compute devices.