Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Compute.DeviceRegistryHelpers Module Reference

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.

Files

file  /__w/Mila/Mila/Mila/Src/Dnn/Compute/Registry/DeviceRegistryHelpers.ixx
 Utility functions for compute device discovery and management.