Ke Wen 2e42671619 [pipelining] Rename to stage.py and schedules.py (#128278) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/128278 Approved by: https://github.com/H-Huang ghstack dependencies: #128273		2024-06-08 04:42:35 +00:00
..
__init__.py	[pipelining] Consolidate test models into a registry (#126114)	2024-05-14 19:11:54 +00:00
model_registry.py	[pipelining] handle param aliasing (#127471)	2024-05-31 01:52:57 +00:00
test_backward.py	[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126)	2024-05-27 14:49:57 +00:00
test_composability.py	[pipelining] Rename to stage.py and schedules.py (#128278)	2024-06-08 04:42:35 +00:00
test_microbatch.py	[pipelining] pipeline() taking microbatch as example input (#128163)	2024-06-07 15:51:53 +00:00
test_pipe.py	[pipelining] pipeline() taking microbatch as example input (#128163)	2024-06-07 15:51:53 +00:00
test_schedule.py	[pipelining] Rename to stage.py and schedules.py (#128278)	2024-06-08 04:42:35 +00:00
test_stage.py	[pipelining] Restore a stage constructor for tracer path (#128273)	2024-06-08 04:42:35 +00:00
test_transformer.py	[pipelining] pipeline() taking microbatch as example input (#128163)	2024-06-07 15:51:53 +00:00
test_unflatten.py	[pipelining] pipeline() taking microbatch as example input (#128163)	2024-06-07 15:51:53 +00:00