mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00

Files

Yuanyuan Chen 3255e7872b Enable all flake8-logging-format rules (#164655 )

These rules are enabled by removing existing suppressions.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164655
Approved by: https://github.com/janeyx99, https://github.com/mlazos

2025-10-19 00:59:28 +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

Enable all flake8-logging-format rules (#164655 )

2025-10-19 00:59:28 +00:00

stage.py

In pipeline parallelism: Use same dtype for receive and send tensor when initializing p2p communication. (#165539 )

2025-10-15 15:05:55 +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.