iron.worker.Worker Class Reference

Inheritance diagram for iron.worker.Worker:

Collaboration diagram for iron.worker.Worker:

Public Member Functions
	__init__ (self, Callable|None core_fn, list fn_args=[], Tile tile=AnyComputeTile, bool while_true=True, int stack_size=None, str allocation_scheme=None, int trace=None, list trace_events=None, bool|None dynamic_objfifo_lowering=None)
list[ObjectFifoHandle]	fifos (self)
list[Buffer]	buffers (self)
None	resolve (self, ir.Location|None loc=None, ir.InsertionPoint|None ip=None)
Public Member Functions inherited from iron.dataflow.endpoint.ObjectFifoEndpoint
Tile|None	tile (self)

Public Attributes
	stack_size
	allocation_scheme
	trace
	trace_events
	core_fn
	fn_args

Protected Attributes
	_tile
	_while_true
	_dynamic_objfifo_lowering
	_fifos
	_buffers
	_barriers
Protected Attributes inherited from iron.dataflow.endpoint.ObjectFifoEndpoint
	_tile

Detailed Description

A task to be run on an AIE compute core.

A Worker takes a ``core_fn`` callable and the arguments it needs (ObjectFIFO handles,
Buffers, Kernels, etc.). Each Worker is placed on a single compute tile, either
explicitly via ``tile`` or automatically by the ``--aie-place-tiles`` compiler pass.

Constructor & Destructor Documentation

__init__()

iron.worker.Worker.__init__	(		self,
		Callable | None	core_fn,
		list	fn_args = [],
		Tile	tile = AnyComputeTile,
		bool	while_true = True,
		int	stack_size = None,
		str	allocation_scheme = None,
		int	trace = None,
		list	trace_events = None,
		bool | None	dynamic_objfifo_lowering = None
	)

Construct a Worker

Args:
    core_fn (Callable | None): The task to run on a core. If None, a busy-loop (`while(true): pass`) core will be generated.
    fn_args (list, optional): Pointers to arguments, which should include all context the core_fn needs to run. Defaults to [].
    tile (Tile, optional): The compute tile for the Worker. Defaults to AnyComputeTile.
    while_true (bool, optional): If true, will wrap the core_fn in a while(true) loop to ensure it runs until reconfiguration. Defaults to True.
    stack_size (int, optional): The stack_size in bytes to be allocated for the worker. Defaults to 1024 bytes.
    allocation_scheme (str, optional): The memory allocation scheme to use for the Worker, either 'basic-sequential' or 'bank-aware'. If None, defaults to bank-aware.
        Will override any allocation scheme set on the tile.
    trace (int, optional): If >0, enable tracing for this worker.
    trace_events (list | None, optional): Custom list of trace events for this worker. Defaults to None.
    dynamic_objfifo_lowering (bool | None, optional): Per-core override for the
        ``aie-objectFifo-stateful-transform`` pass's lowering choice. ``True`` forces
        dynamic (loop-preserving) lowering for this core; ``False`` forces static
        LCM-based unrolling. ``None`` (default) leaves the choice to the compiler's
        global ``--dynamic-objFifos`` flag. Note: the per-core attribute is only
        honored when the global flag is ``false``; when global is ``true`` the
        attribute is ignored. Defaults to None.

Raises:
    ValueError: Parameters are validated.

Reimplemented from iron.dataflow.endpoint.ObjectFifoEndpoint.

Member Function Documentation

buffers()

list[Buffer] iron.worker.Worker.buffers

(

self

)

Returns a list of Buffers given to the Worker via fn_args.

Returns:
    list[Buffer]: Buffer used by the Worker.

fifos()

list[ObjectFifoHandle] iron.worker.Worker.fifos

(

self

)

Returns a list of ObjectFifoHandles given to the Worker via fn_args.

Returns:
    list[ObjectFifoHandle]: ObjectFifoHandles used by the Worker.

resolve()

None iron.worker.Worker.resolve	(		self,
		ir.Location | None	loc = None,
		ir.InsertionPoint | None	ip = None
	)

Member Data Documentation

_barriers

iron.worker.Worker._barriers

protected

_buffers

iron.worker.Worker._buffers

protected

_dynamic_objfifo_lowering

iron.worker.Worker._dynamic_objfifo_lowering

protected

_fifos

iron.worker.Worker._fifos

protected

_tile

iron.worker.Worker._tile

protected

_while_true

iron.worker.Worker._while_true

protected

allocation_scheme

iron.worker.Worker.allocation_scheme

core_fn

iron.worker.Worker.core_fn

fn_args

iron.worker.Worker.fn_args

stack_size

iron.worker.Worker.stack_size

trace

iron.worker.Worker.trace

trace_events

iron.worker.Worker.trace_events

---

The documentation for this class was generated from the following file:

• iron/worker.py