hailin
/
sglang_v0.5.2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
sglang_v0.5.2/pytorch_2.8.0/torch/fx/passes
History
hailin c8e8c1e9ff . 2025-09-20 16:09:34 +08:00
..
backends . 2025-09-20 16:09:34 +08:00
dialect . 2025-09-20 16:09:34 +08:00
infra . 2025-09-20 16:09:34 +08:00
tests . 2025-09-20 16:09:34 +08:00
utils . 2025-09-20 16:09:34 +08:00
README.md . 2025-09-20 16:09:34 +08:00
__init__.py . 2025-09-20 16:09:34 +08:00
_tensorify_python_scalars.py . 2025-09-20 16:09:34 +08:00
annotate_getitem_nodes.py . 2025-09-20 16:09:34 +08:00
fake_tensor_prop.py . 2025-09-20 16:09:34 +08:00
graph_drawer.py . 2025-09-20 16:09:34 +08:00
graph_manipulation.py . 2025-09-20 16:09:34 +08:00
graph_transform_observer.py . 2025-09-20 16:09:34 +08:00
net_min_base.py . 2025-09-20 16:09:34 +08:00
operator_support.py . 2025-09-20 16:09:34 +08:00
param_fetch.py . 2025-09-20 16:09:34 +08:00
pass_manager.py . 2025-09-20 16:09:34 +08:00
reinplace.py . 2025-09-20 16:09:34 +08:00
runtime_assert.py . 2025-09-20 16:09:34 +08:00
shape_prop.py . 2025-09-20 16:09:34 +08:00
split_module.py . 2025-09-20 16:09:34 +08:00
split_utils.py . 2025-09-20 16:09:34 +08:00
splitter_base.py . 2025-09-20 16:09:34 +08:00
tools_common.py . 2025-09-20 16:09:34 +08:00

README.md

FX Pass Infrastructure

This folder contains the pass infrastructure and passes for transforming fx.Graph.

Code Structure

  • infra - Common infrastructure, such as PassManager, PassBase
  • 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
  • backends - Backend specific passes
  • conversion - Conversion passes between dialects