Aaron Orenstein 99dbc5b0e2 PEP585 update - test (#145176) ... See #145101 for details. Pull Request resolved: https://github.com/pytorch/pytorch/pull/145176 Approved by: https://github.com/bobrenjc93		2025-01-22 04:48:28 +00:00
..
artifacts
__init__.py
model_registry.py
schedule_registry.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_backward.py
test_microbatch.py
test_pipe.py
test_schedule.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_schedule_multiproc.py	[Pipelining] Fix PP grad scaling (#144352)	2025-01-14 20:13:17 +00:00
test_stage.py
test_transformer.py
test_unflatten.py