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
_shard	[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408)	2025-02-04 19:07:04 +00:00
_sharded_tensor
_sharding_spec
_symmetric_memory
_tensor
_tools	Turn on mypy for _dynamo/variables/builtin.py (#145552)	2025-01-30 22:21:32 +00:00
algorithms
autograd
benchmarks
checkpoint	[DCP] Remove all-gather of state dict keys (#145998)	2025-02-04 03:16:13 +00:00
elastic	[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408)	2025-02-04 19:07:04 +00:00
examples
fsdp	update _unsafe_set_version_counter to accept lists of tensors (#137921)	2025-02-04 04:51:11 +00:00
launcher
nn
optim	[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408)	2025-02-04 19:07:04 +00:00
pipelining	Remove stage_index_to_group_rank from schedule (#146217)	2025-02-05 21:26:45 +00:00
rpc	[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408)	2025-02-04 19:07:04 +00:00
tensor	[BE][Ez]: ISC001 Auto concatenate implicit one line strings (#146408)	2025-02-04 19:07:04 +00:00
__init__.py
_checkpointable.py
_composable_state.py
_functional_collectives.py	[2/N] Enable ruff F841 on distributed tests (#146132)	2025-02-02 03:44:48 +00:00
_functional_collectives_impl.py
_state_dict_utils.py
argparse_util.py
c10d_logger.py
collective_utils.py
constants.py
CONTRIBUTING.md
device_mesh.py
distributed_c10d.py	[BE]: Enable ruff SLOT checks (#146276)	2025-02-04 19:18:23 +00:00
launch.py
logging_handlers.py
remote_device.py
rendezvous.py
run.py
utils.py