mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
4a12777ffe
We iterate upstream nodes to update partition map. But actually did nothing due to we iterate nodes with reversed topological order https://github.com/pytorch/pytorch/pull/136608/files#diff-f2f9dd3903fd99955732eb694941fea0cb7301a58d59554787f3311d417e5615L193 so that there exists no upstream nodes in assignment. Remove it to reduce for-loop overhead which up to O(N * N) complexity. Pull Request resolved: https://github.com/pytorch/pytorch/pull/146580 Approved by: https://github.com/Skylion007, https://github.com/jerome-habana |
||
|---|---|---|
| .. | ||
| backends | ||
| dialect | ||
| infra | ||
| tests | ||
| utils | ||
| __init__.py | ||
| _tensorify_python_scalars.py | ||
| annotate_getitem_nodes.py | ||
| fake_tensor_prop.py | ||
| graph_drawer.py | ||
| graph_manipulation.py | ||
| graph_transform_observer.py | ||
| net_min_base.py | ||
| operator_support.py | ||
| param_fetch.py | ||
| pass_manager.py | ||
| README.md | ||
| reinplace.py | ||
| runtime_assert.py | ||
| shape_prop.py | ||
| split_module.py | ||
| split_utils.py | ||
| splitter_base.py | ||
| tools_common.py | ||
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
- 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