Ke Wen 8bd26ecf0b [pipelining] test composability with DDP and FSDP (#127066) ... Added to `multigpu` test config, which is run periodically. Pull Request resolved: https://github.com/pytorch/pytorch/pull/127066 Approved by: https://github.com/H-Huang, https://github.com/wconstab ghstack dependencies: #127136, #126931		2024-05-25 04:30:40 +00:00
..
__init__.py	[pipelining] Consolidate test models into a registry (#126114)	2024-05-14 19:11:54 +00:00
model_registry.py	[pipelining] Generalize definition of MultiMLP for testing interleaved schedules (#126927)	2024-05-23 20:26:08 +00:00
test_backward.py	[pipelining] Add pipeline stage test (#126721)	2024-05-22 16:24:51 +00:00
test_chunkspec.py	[pipelining] Add pipeline stage test (#126721)	2024-05-22 16:24:51 +00:00
test_composability.py	[pipelining] test composability with DDP and FSDP (#127066)	2024-05-25 04:30:40 +00:00
test_microbatch.py
test_pipe.py	[pipelining] Consolidate test models into a registry (#126114)	2024-05-14 19:11:54 +00:00
test_schedule.py	[pipelining] Add grad test for interleaved schedules (#126931)	2024-05-25 04:13:28 +00:00
test_stage.py	[pipelining] Generalize definition of MultiMLP for testing interleaved schedules (#126927)	2024-05-23 20:26:08 +00:00
test_transformer.py	[pipelining] Add tests for tracing frontend (#125449)	2024-05-06 20:44:56 +00:00
test_unflatten.py	[pipelining] Follow improvements in export.unflatten (#126217)	2024-05-18 02:24:01 +00:00