pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Yuanyuan Chen b11593c31b [8/N] Apply ruff UP035 rule (#165214 )

This is follow-up of #164653 to continue applying `UP035` fixes. The purpose is to finally enable this rule.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165214
Approved by: https://github.com/ezyang

2025-10-15 03:18:57 +00:00

backends

[2/N] Simplify "in" operation for containers of a single item (#164323 )

2025-10-01 05:39:11 +00:00

dialect

PEP585 update - torch/fx (#145166 )

2025-01-20 18:11:54 +00:00

infra

[8/N] Apply ruff UP035 rule (#165214 )

2025-10-15 03:18:57 +00:00

tests

[BE][Easy] enable PYFMT for torch.fx (#138443 )

2024-10-21 19:15:49 +00:00

utils

[8/N] Apply ruff UP035 rule (#165214 )

2025-10-15 03:18:57 +00:00

__init__.py

[compile] Regional inductor compilation with fx.annotate (#164776 )

2025-10-13 22:22:20 +00:00

_tensorify_python_scalars.py

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +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

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

graph_drawer.py

Add pyrefly suppressions (#164748 )

2025-10-07 17:31:18 +00:00

graph_manipulation.py

[BE][14/16] fix typos in torch/ (torch/fx/) (#156604 )

2025-07-02 22:55:29 +00:00

graph_transform_observer.py

[8/N] Apply ruff UP035 rule (#165214 )

2025-10-15 03:18:57 +00:00

net_min_base.py

[8/N] Apply ruff UP035 rule (#165214 )

2025-10-15 03:18:57 +00:00

operator_support.py

PEP585 update - torch/fx (#145166 )

2025-01-20 18:11:54 +00:00

param_fetch.py

[8/N] Apply ruff UP035 rule (#165214 )

2025-10-15 03:18:57 +00:00

pass_manager.py

[8/N] Apply ruff UP035 rule (#165214 )

2025-10-15 03:18:57 +00:00

README.md

Fix xrefs (#151888 )

2025-04-25 21:27:27 +00:00

regional_inductor.py

[compile] Regional inductor compilation with fx.annotate (#164776 )

2025-10-13 22:22:20 +00:00

reinplace.py

[8/N] Apply ruff UP035 rule (#165214 )

2025-10-15 03:18:57 +00:00

runtime_assert.py

Pyrefly suppressions 6/n (#164877 )

2025-10-08 02:30:57 +00:00

shape_prop.py

Use tuples to have a deterministic ordering. (#164851 )

2025-10-08 02:12:03 +00:00

split_module.py

[8/N] Apply ruff UP035 rule (#165214 )

2025-10-15 03:18:57 +00:00

split_utils.py

[tgif] fix getattr_recursive with ModuleList (#161204 )

2025-08-25 10:08:47 +00:00

splitter_base.py

[Lowering] Fix the edge case of empty subgraph split due to dataclass node (#161716 )

2025-09-10 21:23:42 +00:00

tools_common.py

PEP585 update - torch/fx (#145166 )

2025-01-20 18:11:54 +00: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
- 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
backends - Backend specific passes
conversion - Conversion passes between dialects