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
__init__.py
model_registry.py
schedule_registry.py	Remove stage_index_to_group_rank from schedule (#146217)	2025-02-05 21:26:45 +00:00
test_backward.py
test_microbatch.py
test_pipe.py
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
test_transformer.py
test_unflatten.py