Skip to content

Python API

IRON's Python API, generated from its docstrings. It sits on mlir-aie's aie.iron, whose own reference is on the MLIR-AIE site.

Page Module What it holds
Graphs iron Graph, its compiled form, and what a body() declares with
Functional iron.functional functions for a graph's body, under torch.nn.functional's names
Operators iron.operators the operator library, and the FastFlowLM ports
Writing an operator iron.common Operator, its operands and values, the elementwise templates
Testing and tracing iron.common.testing how an operator declares its tests, and the device test harness
Language models iron.lm what every language model shares
Tuning iron.tune timing operator settings and writing profiles