iron.runtime.task.InlineOpRuntimeTask Class Reference

Inheritance diagram for iron.runtime.task.InlineOpRuntimeTask:

Collaboration diagram for iron.runtime.task.InlineOpRuntimeTask:

Public Member Functions
def	__init__ (self, Callable fn, list args, RuntimeTaskGroup|None task_group=None)
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
	_fn
	_args

Detailed Description

An InlineOpRuntimeTask is a way of submitting arbitrary operations to a runtime that are defined
in a lower-level style of IRON. This can be especially useful for tracing.

Constructor & Destructor Documentation

__init__()

def iron.runtime.task.InlineOpRuntimeTask.__init__	(		self,
		Callable	fn,
		list	args,
		RuntimeTaskGroup | None	task_group = None
	)

Construct an InlineOpRuntimeTask.

Args:
    fn (Callable): The function that will generate ops. It will be run within an MLIR module context.
    args (list): The arguments for the task: this should included objects such as Buffers used by the function.
    task_group (RuntimeTaskGroup | None, optional): The TaskGroup to associated these operation with. Defaults to None.

Member Function Documentation

resolve()

None iron.runtime.task.InlineOpRuntimeTask.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.

Member Data Documentation

_args

iron.runtime.task.InlineOpRuntimeTask._args

private

_fn

iron.runtime.task.InlineOpRuntimeTask._fn

private

---

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

• iron/runtime/task.py