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
..
_composable	Refactoring Distributed test cases to be device agnostic [1/n] (#145222)	2025-02-05 18:47:09 +00:00
_shard	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
_tools	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
algorithms	Enable ruff F841 on distributed tests (#146131)	2025-02-01 03:06:16 +00:00
bin
checkpoint	[DCP] Remove all-gather of state dict keys (#145998)	2025-02-04 03:16:13 +00:00
elastic	Enable ruff F841 on distributed tests (#146131)	2025-02-01 03:06:16 +00:00
flight_recorder	Revert "Use absolute path path.resolve() -> path.absolute() (#129409)"	2025-01-04 14:17:20 +00:00
fsdp	[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408)	2025-02-04 19:07:04 +00:00
launcher	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
nn/jit	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
optim	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
pipelining	Remove stage_index_to_group_rank from schedule (#146217)	2025-02-05 21:26:45 +00:00
rpc
tensor	[2/N] Enable ruff F841 on distributed tests (#146132)	2025-02-02 03:44:48 +00:00
argparse_util_test.py
test_backends.py
test_c10d_common.py	[BE][Ez]: FURB148 - remove useless enumerate calls (#145619)	2025-01-24 23:37:15 +00:00
test_c10d_functional_native.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_c10d_gloo.py	[ROCm] Enable post-merge trunk workflow on MI300 runners; skip and fix MI300 related failed tests (#143673)	2025-01-09 05:18:57 +00:00
test_c10d_logger.py
test_c10d_nccl.py	[c10d][NCCL] Implement ncclCommInitRankScalable (merging #136789) (#144794)	2025-01-31 22:39:56 +00:00
test_c10d_object_collectives.py
test_c10d_ops_nccl.py
test_c10d_pypg.py
test_c10d_spawn.py
test_c10d_spawn_gloo.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_c10d_spawn_nccl.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_c10d_spawn_ucc.py	PEP585 update - test (#145176)	2025-01-22 04:48:28 +00:00
test_c10d_ucc.py	XFAIL test_save_load_checkpoint (#144927)	2025-01-16 07:31:56 +00:00
test_collective_utils.py
test_composability.py	composability test cleanup (#145011)	2025-01-18 04:37:12 +00:00
test_compute_comm_reordering.py
test_control_collectives.py
test_data_parallel.py
test_device_mesh.py
test_distributed_spawn.py
test_dynamo_distributed.py	dynamo: fsdp throw unimplemented vs attribute error (#146188)	2025-02-04 21:45:55 +00:00
test_fake_pg.py	[BE][Ez]: FURB148 - remove useless enumerate calls (#145619)	2025-01-24 23:37:15 +00:00
test_functional_api.py
test_inductor_collectives.py
test_launcher.py
test_multi_threaded_pg.py
test_nccl.py
test_pg_wrapper.py
test_store.py	[BE][CI] bump ruff to 0.8.4 (#143753)	2024-12-24 12:24:10 +00:00
test_symmetric_memory.py	[SymmetricMemory] fix an issue where rendezvous is performed with wrong device context when torch.cuda.set_device() is not callled (#144886)	2025-01-28 01:43:37 +00:00