OSSForks/pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00

History

PyTorch MergeBot 3443627e07 Revert "[BE]: Enable RUFF TRY400 rule - log.exception (#153473 )" This reverts commit `4f4ecc583e`. Reverted https://github.com/pytorch/pytorch/pull/153473 on behalf of https://github.com/jeanschmidt due to seems to have broken internal signals, @albanD may I count on you to help the author merge his PR? D74837988 ([comment](https://github.com/pytorch/pytorch/pull/153473#issuecomment-2886017075))		2025-05-16 08:29:26 +00:00
..
__init__.py	[pipelining] Add ZBV schedule (#142084 )	2024-12-11 02:00:57 +00:00
_backward.py	[BE][PYFMT] migrate PYFMT for `torch.{distributed,distributions}` to `ruff format` (#144547 )	2025-02-28 07:35:56 +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	[Codemod][AddExplicitStrictExportForTrainingInferenceArg] caffe2/ (#149595 )	2025-04-03 23:50:13 +00:00
_schedule_visualizer.py	[PP] Add schedule visualizer (#150347 )	2025-04-15 00:38:18 +00:00
_unflatten.py	PEP585 update - torch/distributed (#145164 )	2025-01-21 04:23:29 +00:00
_utils.py	PEP585: More UP006 fixes (#146392 )	2025-02-20 06:18:13 +00:00
microbatch.py	[BE][PYFMT] migrate PYFMT for `torch.{distributed,distributions}` to `ruff format` (#144547 )	2025-02-28 07:35:56 +00:00
README.md
schedules.py	Revert "[BE]: Enable RUFF TRY400 rule - log.exception (#153473 )"	2025-05-16 08:29:26 +00:00
stage.py	[PP] Optimize memory usage by releasing output memory earlier (#153383 )	2025-05-13 14:42:38 +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.