iron.runtime.dmatask.DMATask Class Reference

Inheritance diagram for iron.runtime.dmatask.DMATask:

Collaboration diagram for iron.runtime.dmatask.DMATask:

Public Member Functions
def	__init__ (self, ObjectFifoHandle object_fifo, RuntimeData rt_data, TensorAccessPattern tap, RuntimeTaskGroup|None task_group=None, bool wait=False)
bool	will_wait (self)
ObjectFifoHandle	fifo (self)
def	task (self)
None	resolve (self, ir.Location|None loc=None, ir.InsertionPoint|None ip=None)
Public Member Functions inherited from iron.runtime.task.RuntimeTask
def	__init__ (self, RuntimeTaskGroup|None task_group=None)
RuntimeTaskGroup|None	task_group (self)

Private Attributes
	_object_fifo
	_rt_data
	_tap
	_wait
	_task

Constructor & Destructor Documentation

__init__()

def iron.runtime.dmatask.DMATask.__init__	(		self,
		ObjectFifoHandle	object_fifo,
		RuntimeData	rt_data,
		TensorAccessPattern	tap,
		RuntimeTaskGroup | None	task_group = None,
		bool	wait = False
	)

A RuntimeTask that will resolve to a DMA Operation.

Args:
    object_fifo (ObjectFifoHandle): The ObjectFifoHandle associated with the operation
    rt_data (RuntimeData): The Runtime buffer associated with the operation.
    tap (TensorAccessPattern): The access pattern associated with the operation.
    task_group (RuntimeTaskGroup | None, optional): The task group associated with the operation. Defaults to None.
    wait (bool, optional): Whether this task should conclude with a call to await or a call to free. Defaults to False.

Member Function Documentation

fifo()

ObjectFifoHandle iron.runtime.dmatask.DMATask.fifo

(

self

)

The ObjectFifoHandle associated with this task.

resolve()

None iron.runtime.dmatask.DMATask.resolve	(		cls,
		ir.Location | None	loc = None,
		ir.InsertionPoint | None	ip = None
	)

Resolve the current object into one or more MLIR operations.
Should only be called within an MLIR context.

Args:
    loc (ir.Location | None, optional): Location is used by MLIR object during construction in some cases. Defaults to None.
    ip (ir.InsertionPoint | None, optional): InsertionPoint is used by MLIR object during construction in some cases. Defaults to None.

Reimplemented from iron.runtime.task.RuntimeTask.

task()

def iron.runtime.dmatask.DMATask.task

(

self

)

The MLIR op generated by resolve(). This handle is used after resolution for await or free operations as
dictated by a RuntimeTaskGroup Finish operation

will_wait()

bool iron.runtime.dmatask.DMATask.will_wait

(

self

)

If this tasks shoudl conclude with an await operation.

Member Data Documentation

_object_fifo

iron.runtime.dmatask.DMATask._object_fifo

private

_rt_data

iron.runtime.dmatask.DMATask._rt_data

private

_tap

iron.runtime.dmatask.DMATask._tap

private

_task

iron.runtime.dmatask.DMATask._task

private

_wait

iron.runtime.dmatask.DMATask._wait

private

---

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

• iron/runtime/dmatask.py