OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
df458be4e5
pytorch/torch/fx/passes
History
cyy df458be4e5 [4/N] Apply py39 ruff and pyupgrade fixes (#143257) 
```torch/fx/passes/annotate_getitem_nodes.py``` was changed to support the new type hinting annotations.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/143257
Approved by: https://github.com/justinchuby, https://github.com/albanD
2025-01-04 10:47:51 +00:00
..
backends [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00
dialect [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00
infra Remove assert from partitioner.py (#143376) 2024-12-18 06:08:19 +00:00
tests [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00
utils Fix get_source_partitions when weights are tied (#142446) 2025-01-04 04:28:20 +00:00
__init__.py [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00
_tensorify_python_scalars.py Revert "refactor tensorify restart logic to use sources (#141517)" (#143623) 2024-12-20 15:38:34 +00:00
annotate_getitem_nodes.py [4/N] Apply py39 ruff and pyupgrade fixes (#143257) 2025-01-04 10:47:51 +00:00
fake_tensor_prop.py [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00
graph_drawer.py Fix type-safety of torch.nn.Module instances (#141240) 2024-11-22 00:05:05 +00:00
graph_manipulation.py [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00
graph_transform_observer.py Class rename (#139490) 2024-11-02 00:10:17 +00:00
net_min_base.py [3/x][fx minimizer] Support all_outputs in minimizer (#139774) 2024-11-13 22:56:42 +00:00
operator_support.py [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00
param_fetch.py [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00
pass_manager.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
README.md [Doc] Fix typo in torch/fx/passes/README.md (#134078) 2024-08-21 18:35:50 +00:00
reinplace.py [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00
runtime_assert.py Unbacked SymInt fixes for subclasses + data-dependent slice() bounds (non-dynamic) (#143526) 2024-12-19 18:46:36 +00:00
shape_prop.py Fix stride in TensorMetadata to always be a Tuple[int, ...] (#141106) 2024-11-21 14:52:36 +00:00
split_module.py [fx] make split_module work with keep_original_order=True and no-op graph (#141340) 2024-11-24 06:41:30 +00:00
split_utils.py [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00
splitter_base.py [IG] Avoid generation of empty merge cpu submodule by splitter v2 (#140794) 2024-11-16 01:49:03 +00:00
tools_common.py [BE][Easy] enable PYFMT for torch.fx (#138443) 2024-10-21 19:15:49 +00:00

README.md

FX Pass Infrastructure

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

Code Structure