PyTorch ExecuTorch Export IR Specification
Properties | |
---|---|
authors | PyTorch - Functionalization in PyTorch - Everything you need to know |
year | 2024 |
url | https://pytorch.org/executorch/main/ir-exir.html |
The Exported IR is a specification that consists of the following parts:
- A definition of computation graph model.
- Set of operators allowed in the graph.
A dialect also provides further constraints meant for a specific purpose or stage in some compilation phase. Some dialects are:
- aten dialect
- edge dialect
- backend dialect
Executorch compilation first exports to aten, then to edge and finally to backend.
Aten Dialect¶
- PyTorch Functionalization is performed, removing any tensor aliases and mutations, and allowing for more flexible graph transformations to be made.