mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
Pull Request resolved: https://github.com/pytorch/pytorch/pull/165282 Approved by: https://github.com/ezyang, https://github.com/mlazos
FX Pass Infrastructure
This folder contains the pass infrastructure and passes for transforming fx.Graph.
Code Structure
- infra - Common infrastructure, such as PassManager, PassBase
- partitioner.py - backend agnostic FX graph partitioner
- utils - Utility classes and functions
- common.py - common utility functions
- fuser_utils.py - utility functions for fusing list of nodes into a single node
- dialect - dialect specific passes
- common - common passes that can be shared by all dialects
- cse_pass.py - a CSE pass
- common - common passes that can be shared by all dialects
- backends - Backend specific passes
- conversion - Conversion passes between dialects