This folder contains the pass infrastructure and passes for transforming fx.Graph.
- 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_utis.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
- aten - aten dialect specific passes
- prims - prim dialect specific passes
- common - common passes that can be shared by all dialects
- backends - Backend specific passes
- nvfuser - passes for nvfuser
- operator_support.py - nvFuser supported ops
- nvfuser - passes for nvfuser
- conversion - Conversion passes between dialects