Inheritance diagram for iron.runtime.task.InlineOpRuntimeTask:
Collaboration diagram for iron.runtime.task.InlineOpRuntimeTask:
Public Member Functions | |
| __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 | |
| RuntimeTaskGroup|None | task_group (self) |
Protected Attributes | |
| _fn | |
| _args | |
| Protected Attributes inherited from iron.runtime.task.RuntimeTask | |
| _task_group | |
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__()
| 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.
Reimplemented from iron.runtime.task.RuntimeTask.
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
|
protected |
◆ _fn
|
protected |
The documentation for this class was generated from the following file:
- iron/runtime/task.py
