iron.device.device.NPUBase Class Reference

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)
int	get_num_source_switchbox_connections (self, Tile t)
int	get_num_dest_switchbox_connections (self, Tile t)
int	get_num_source_shim_mux_connections (self, Tile t)
int	get_num_dest_shim_mux_connections (self, Tile t)
int	get_num_connections (self, Tile tile, bool output)
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

Protected Attributes
	_device
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_num_connections()

int iron.device.device.NPUBase.get_num_connections	(		self,
		Tile	tile,
		bool	output
	)

get_num_dest_shim_mux_connections()

int iron.device.device.NPUBase.get_num_dest_shim_mux_connections	(		self,
		Tile	t
	)

Returns number of DMA dest ports in the shim mux for the given tile on the device.

Returns:
    int: Number of DMA dest ports.

Reimplemented from iron.device.device.Device.

get_num_dest_switchbox_connections()

int iron.device.device.NPUBase.get_num_dest_switchbox_connections	(		self,
		Tile	t
	)

Returns number of DMA dest ports in the switchbox for the given tile on the device.

Returns:
    int: Number of DMA dest ports.

Reimplemented from iron.device.device.Device.

get_num_source_shim_mux_connections()

int iron.device.device.NPUBase.get_num_source_shim_mux_connections	(		self,
		Tile	t
	)

Returns number of DMA source ports in the shim mux for the given tile on the device.

Returns:
    int: Number of DMA source ports.

Reimplemented from iron.device.device.Device.

get_num_source_switchbox_connections()

int iron.device.device.NPUBase.get_num_source_switchbox_connections	(		self,
		Tile	t
	)

Returns number of DMA source ports in the switchbox for the given tile on the device.

Returns:
    int: Number of DMA source ports.

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

_device

iron.device.device.NPUBase._device

protected

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