Howard Huang 9b6d680131 Remove stage_index_to_group_rank from schedule (#146217) ... This PR allows schedules loaded via CSV to automatically set their `stage_index_to_group_rank ` and removes the `stage_index_to_group_rank ` argument from the `PipelineScheduleMulti` constructor Pull Request resolved: https://github.com/pytorch/pytorch/pull/146217 Approved by: https://github.com/wconstab ghstack dependencies: #146193		2025-02-05 21:26:45 +00:00
..
artifacts	[Pipelining] Support separate dI / dW and V-schedules (#131762)	2024-10-31 22:49:45 +00:00
__init__.py	[pipelining] Consolidate test models into a registry (#126114)	2024-05-14 19:11:54 +00:00
model_registry.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
schedule_registry.py	Remove stage_index_to_group_rank from schedule (#146217)	2025-02-05 21:26:45 +00:00
test_backward.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
test_microbatch.py	[pipelining] pipeline() taking microbatch as example input (#128163)	2024-06-07 15:51:53 +00:00
test_pipe.py	Add None return type to init -- tests rest (#132376)	2024-08-01 15:44:51 +00:00
test_schedule.py	Remove stage_index_to_group_rank from schedule (#146217)	2025-02-05 21:26:45 +00:00
test_schedule_multiproc.py	Remove stage_index_to_group_rank from schedule (#146217)	2025-02-05 21:26:45 +00:00
test_stage.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00
test_transformer.py	[pipelining] pipeline() taking microbatch as example input (#128163)	2024-06-07 15:51:53 +00:00
test_unflatten.py	Fix unused Python variables in test/[a-d]* (#134665)	2024-12-13 22:13:12 +00:00