| Name | Date | Size | #Lines | LOC | ||
|---|---|---|---|---|---|---|
| .. | - | - | ||||
| backends/ | H | 25-Apr-2025 | - | 59 | 38 | |
| dialect/ | H | 25-Apr-2025 | - | 116 | 79 | |
| infra/ | H | 25-Apr-2025 | - | 716 | 525 | |
| tests/ | H | 25-Apr-2025 | - | 60 | 44 | |
| utils/ | H | 25-Apr-2025 | - | 1,008 | 686 | |
| README.md | H A D | 25-Apr-2025 | 1.1 KiB | 21 | 17 | |
| __init__.py | H A D | 25-Apr-2025 | 330 | 13 | 12 | |
| annotate_getitem_nodes.py | H A D | 25-Apr-2025 | 1.9 KiB | 45 | 37 | |
| fake_tensor_prop.py | H A D | 25-Apr-2025 | 2.7 KiB | 71 | 56 | |
| graph_drawer.py | H A D | 25-Apr-2025 | 17.2 KiB | 444 | 367 | |
| graph_manipulation.py | H A D | 25-Apr-2025 | 3.9 KiB | 112 | 89 | |
| graph_transform_observer.py | H A D | 25-Apr-2025 | 2.9 KiB | 92 | 71 | |
| net_min_base.py | H A D | 25-Apr-2025 | 33.6 KiB | 925 | 740 | |
| operator_support.py | H A D | 25-Apr-2025 | 7.5 KiB | 216 | 154 | |
| param_fetch.py | H A D | 25-Apr-2025 | 3.4 KiB | 67 | 54 | |
| pass_manager.py | H A D | 25-Apr-2025 | 7 KiB | 255 | 196 | |
| reinplace.py | H A D | 25-Apr-2025 | 32.6 KiB | 676 | 417 | |
| runtime_assert.py | H A D | 25-Apr-2025 | 27.4 KiB | 606 | 443 | |
| shape_prop.py | H A D | 25-Apr-2025 | 7.2 KiB | 197 | 149 | |
| split_module.py | H A D | 25-Apr-2025 | 23.9 KiB | 576 | 428 | |
| split_utils.py | H A D | 25-Apr-2025 | 11 KiB | 304 | 195 | |
| splitter_base.py | H A D | 25-Apr-2025 | 32.6 KiB | 899 | 696 | |
| tools_common.py | H A D | 25-Apr-2025 | 10.9 KiB | 304 | 234 |
README.md
1## FX Pass Infrastructure 2This folder contains the pass infrastructure and passes for transforming fx.Graph. 3 4 5## Code Structure 6 7* [infra](infra) - Common infrastructure, such as PassManager, PassBase 8 * [partitioner.py](infra/partitioner.py) - backend agnostic FX graph partitioner 9* [utils](utils) - Utility classes and functions 10 * [common.py](utils/common.py) - common utility functions 11 * [fuser_utils.py](utils/fuser_utils.py) - utility functions for fusing list of nodes into a single node 12* [dialect](dialect) - dialect specific passes 13 * [common](dialect/common) - common passes that can be shared by all dialects 14 * [cse_pass.py](dialect/common/cse_pass.py) - a CSE pass 15 * [aten](dialect/aten) - aten dialect specific passes 16 * [prims](dialect/prims) - prim dialect specific passes 17* [backends](backends) - Backend specific passes 18 * [nvfuser](backends/nvfuser) - passes for nvfuser 19 * [operator_support.py](backends/nvfuser/operator_support.py) - nvFuser supported ops 20* [conversion](conversion) - Conversion passes between dialects 21