Inheritance diagram for iron.device.device.NPUBase:
Collaboration diagram for iron.device.device.NPUBase:
Public Member Functions | |
| None | __init__ (self, AIEDevice device) |
| list[Tile] | get_shim_tiles (self) |
| list[Tile] | get_mem_tiles (self) |
| list[Tile] | get_compute_tiles (self) |
 Public Member Functions inherited from iron.device.device.Device | |
| int | rows (self) |
| int | cols (self) |
| None | resolve_tile (self, Tile tile, ir.Location|None loc=None, ir.InsertionPoint|None ip=None) |
| Public Member Functions inherited from iron.resolvable.Resolvable | |
| None | resolve (cls, ir.Location|None loc=None, ir.InsertionPoint|None ip=None) |
Public Attributes | |
| cols | |
| rows | |
Additional Inherited Members | |
| Protected Attributes inherited from iron.device.device.Device | |
| _device | |
Detailed Description
A base class which can be used to create other device specific classes. This class is abstract because it does not implement resolve() This class makes some assumptions: * The 0th tile in each column is a shim tile * The 1st tile in each column is a mem tile * The 2nd+ tiles in each column are compute tiles
Constructor & Destructor Documentation
◆ __init__()
| None iron.device.device.NPUBase.__init__ | ( | self, | |
| AIEDevice | device | ||
| ) |
Initialize a device based on the AIEDevice.
Args:
device (AIEDevice): aie device
Reimplemented from iron.device.device.Device.
Member Function Documentation
◆ get_compute_tiles()
| list[Tile] iron.device.device.NPUBase.get_compute_tiles | ( | self | ) |
Returns a list of all compute tiles on the device.
Returns:
list[Tile]: A list of compute tiles.
Reimplemented from iron.device.device.Device.
◆ get_mem_tiles()
| list[Tile] iron.device.device.NPUBase.get_mem_tiles | ( | self | ) |
Returns a list of all mem tiles on the device.
Returns:
list[Tile]: A list of mem tiles.
Reimplemented from iron.device.device.Device.
◆ get_shim_tiles()
| list[Tile] iron.device.device.NPUBase.get_shim_tiles | ( | self | ) |
Returns a list of all shim tiles on the device.
Returns:
list[Tile]: A list of shim tiles.
Reimplemented from iron.device.device.Device.
Member Data Documentation
◆ cols
| iron.device.device.NPUBase.cols |
◆ rows
| iron.device.device.NPUBase.rows |
The documentation for this class was generated from the following file:
- iron/device/device.py
