mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
1cc62c2cb9
Preview: https://docs-preview.pytorch.org/pytorch/pytorch/157750/export.html Changes: * Rename draft_export.md -> export.draft_export.md for consistency. * Removed non-strict section in export, instead pointed to programming model doc. * Extended "Expressing Dynamism" section to include Dim hints, ShapeCollection, and AdditionalInputs. * Removed Specialization section in favor of programming model doc * Added pt2 archive doc * Cleaned up sidebar Pull Request resolved: https://github.com/pytorch/pytorch/pull/157750 Approved by: https://github.com/pianpwk
41 lines
1.3 KiB
Markdown
41 lines
1.3 KiB
Markdown
(torch.compiler_ir)=
|
|
|
|
# IRs
|
|
|
|
PyTorch 2.0 offers two set of IRs for backends to interface with: Core Aten IR and Prims IR.
|
|
|
|
## Core Aten IR
|
|
|
|
Core aten ops is the core subset of aten operators that can be used to compose other operators.
|
|
Core aten IR is fully functional, and there is no `inplace` or `_out` variants in this opset.
|
|
In contrast to Prims IR, core aten ops reuses the existing aten ops in "native_functions.yaml",
|
|
and it doesn't further decompose ops into explicit type promotion and broadcasting ops.
|
|
This opset is designed to serve as the functional IR to interface with backends.
|
|
|
|
```{warning}
|
|
This opset is still under active development, more ops will be added in the future.
|
|
```
|
|
|
|
```{csv-table}
|
|
:file: ../build/ir/aten_ops.csv
|
|
:widths: auto
|
|
:header-rows: 1
|
|
```
|
|
|
|
## Prims IR
|
|
|
|
Prims IR is a set of primitive operators that can be used to compose other operators.
|
|
Prims IR is a lower level opset than core aten IR, and it further decomposes ops into explicit
|
|
type promotion and broadcasting ops: prims.convert_element_type and prims.broadcast_in_dim.
|
|
This opset is designed to interface with compiler backends.
|
|
|
|
```{warning}
|
|
This opset is still under active development, more ops will be added in the future.
|
|
```
|
|
|
|
```{csv-table}
|
|
:file: ../build/ir/prims_ops.csv
|
|
:widths: auto
|
|
:header-rows: 1
|
|
```
|