mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-04 08:00:58 +08:00
2b138e4f7d
Pull Request resolved: https://github.com/pytorch/pytorch/pull/108783 Approved by: https://github.com/avikchaudhuri, https://github.com/gmagogsfm
40 lines
1.3 KiB
ReStructuredText
40 lines
1.3 KiB
ReStructuredText
.. _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
|