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 |