DevManager
- class DevManager(devices: Union[List[int], int], backend_name='')
Describes the device set on which the target model runs. For multi-device models, initialization must be performed using DevManager.
The DevManager class provides device management functionalities, including creating device manager instances and retrieving the number of devices. During initialization, it checks the status of the device manager, and if initialization fails, an exception will be raised.
Initializes a DevManager instance.
- Parameters:
devices -- Device ID or a list of device IDs. Can be a single device ID (int) or a list of device IDs (List[int]).
backend_name -- The backend name, default is an empty string.
- Raises:
RuntimeError -- If the device manager fails to initialize, a runtime error is raised.
Methods
Initializes a DevManager instance.
Retrieve the number of available devices.
Verify the accuracy of the device id.