Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
DeviceRegistryHelpers.ixx File Reference

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.

Detailed Description

Utility functions for compute device discovery and management.

Small helpers for listing and selecting devices via DeviceRegistry.