maccel.accelerator.Accelerator Class Reference#
|
Runtime Library v0.30
Mobilint SDK qb
|
maccel.accelerator.Accelerator Class Reference
Represents an accelerator, i.e., an NPU, used for executing models. More...
Public Member Functions | |
| __init__ (self, int dev_no=0) | |
| Creates an Accelerator object for a specific device number. | |
| List[CoreId] | get_available_cores (self) |
| Retrieves a list of available NPU cores. | |
Protected Attributes | |
| _accelerator = _cMaccel.Accelerator(dev_no) | |
Detailed Description
Represents an accelerator, i.e., an NPU, used for executing models.
Definition at line 17 of file accelerator.py.
Constructor & Destructor Documentation
◆ __init__()
| maccel.accelerator.Accelerator.__init__ | ( | self, | |
| int | dev_no = 0 ) |
Creates an Accelerator object for a specific device number.
The dev_no parameter represents the device number. For example, on Linux, if an ARIES NPU is attached as /dev/aries0, the device number is 0.
- Parameters
-
dev_no The device number to associate with the Accelerator.
Definition at line 20 of file accelerator.py.
Member Function Documentation
◆ get_available_cores()
| List[CoreId] maccel.accelerator.Accelerator.get_available_cores | ( | self | ) |
Retrieves a list of available NPU cores.
An available core is one that can be allocated for newly created Model objects.
- Note
- Availability checks are only supported on Linux. On Windows, this function returns all NPU cores without checking availability.
- Returns
- A list containing the IDs of available cores.
Definition at line 31 of file accelerator.py.
Member Data Documentation
◆ _accelerator
|
protected |
Definition at line 29 of file accelerator.py.
The documentation for this class was generated from the following file:
Generated by
1.14.0