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_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
• backends - Backend specific passes
  • nvfuser - passes for nvfuser
    • operator_support.py - nvFuser supported ops
• conversion - Conversion passes between dialects