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

__init__.py

[PP] Add DualPipeV schedule (#159591 )

2025-08-14 14:58:35 +00:00

_backward.py

[2/N] More ruff SIM fixes (#165031 )

2025-10-14 14:22:54 +00:00

_debug.py

remove allow-untyped-defs from torch/distributed/pipelining/_debug.py (#143871 )

2024-12-27 01:20:26 +00:00

_IR.py

[2/N] More ruff SIM fixes (#165031 )

2025-10-14 14:22:54 +00:00

_schedule_visualizer.py

Add pyrefly suppressions to torch/distributed (7/n) (#165002 )

2025-10-09 04:08:25 +00:00

_unflatten.py

PEP585 update - torch/distributed (#145164 )

2025-01-21 04:23:29 +00:00

_utils.py

[PP] Support OVERLAP_F_B computation type (#158978 )

2025-08-01 20:22:30 +00:00

microbatch.py

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

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

README.md

[pipelining] redirect README to document (#128205 )

2024-06-07 19:34:52 +00:00

schedules.py

[PP] Fix edge case with FSDP when stages_per_rank > 3 (#165467 )

2025-10-15 01:53:04 +00:00

stage.py

[PP] move FSDP reduce scatters to end of step (#165106 )

2025-10-12 13:28:02 +00:00

README.md

Pipeline Parallelism for PyTorch

torch.distributed.pipelining is a package for implementing pipeline parallelism on your model.

Our documentation is available here.