|
Mila 0.13.48
Deep Neural Network Library
|
Utility functions for compute device discovery and management. More...
#include <vector>#include <string>import Compute.DeviceType;import Compute.DeviceId;import Compute.DeviceRegistry;Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
Functions | |
| DeviceId | Mila::Dnn::Compute::getBestDevice (DeviceType type, bool preferMemory=false) |
| Gets the best DeviceId of a specific type based on performance characteristics. | |
| std::size_t | Mila::Dnn::Compute::getDeviceCount (DeviceType type) noexcept |
| Count instantiated devices of the given DeviceType. | |
| std::vector< std::string > | Mila::Dnn::Compute::listDevicesByName () |
| Lists all available compute devices by name. | |
| std::vector< std::string > | Mila::Dnn::Compute::listDevicesByType (DeviceType type) |
| Lists compute devices of a specific type. | |
Utility functions for compute device discovery and management.
Small helpers for listing and selecting devices via DeviceRegistry.