pytorch/torch/distributed
2023-12-20 06:40:32 +00:00
..
_composable [replicate] Simplify replicate() init logic and remove unnecessary variables in _ReplicateState (#113679) 2023-11-28 00:55:36 +00:00
_shard Updated docs for deprecated torch.set_default_tensor_type (#115041) 2023-12-07 16:17:36 +00:00
_sharded_tensor
_sharding_spec
_spmd [DeviceMesh] Make our mesh_dim kwarg naming consistent (#114707) 2023-11-29 19:43:23 +00:00
_tensor [DTensor] Make DTensor from_local backward partial() to replicate() pass through (#115967) 2023-12-19 00:16:10 +00:00
_tools fix: docstring error in torch/distributed module (#113241) 2023-11-09 19:10:20 +00:00
algorithms Fix docstring errors in default_hooks.py, optimizer_overlap.py, checkpoint_wrapper.py, copy.py, benchmark_ddp_rpc.py, utils.py, dependency.py, phony.py, pipeline.py, checkpoint.py, worker.py, batchnorm.py, quantization.py (#113511) 2023-11-14 18:52:41 +00:00
autograd
benchmarks [BE]: ruff apply rule PLW1510 to find silent subprocess errors (#113644) 2023-11-14 20:59:40 +00:00
checkpoint [DCP][BE] Apply ufmt to DCP and turn on lintrunner for DCP (#115302) 2023-12-13 10:32:36 +00:00
elastic Switch env variable use in test harnesses to the non-deprecated names to fix warnings (#114880) 2023-12-01 20:08:23 +00:00
examples
fsdp [fsdp][torch.compile] FSDP changes (#115497) 2023-12-19 18:44:36 +00:00
launcher [TorchElastic] Add option to configure log prefix for each rank (#112357) 2023-11-08 01:00:26 +00:00
nn fix docstring issues in torch.distributed (#113337) 2023-11-13 19:37:29 +00:00
optim Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830) 2023-11-17 18:24:21 +00:00
pipeline Doc: Add and fix docstrings for torch.distributed files (#112735) 2023-11-15 00:49:07 +00:00
rpc [BE]: Remove useless lambdas (#113602) 2023-11-14 20:06:48 +00:00
tensor Fix ColwiseParallel typo (#116151) 2023-12-20 06:40:32 +00:00
__init__.py Fix torch.distributed.breakpoint (#115705) 2023-12-13 20:33:56 +00:00
_composable_state.py Fix docstring errors in _composable_state.py, remote_device.py, value_ranges.py, utils.py, run.py, rendezvous.py, launch.py, argparse_util.py, __init__.py, _cycles.py (#112953) 2023-11-08 01:13:09 +00:00
_functional_collectives_impl.py [funcol][BE] Apply ufmt to _functional_collectives.py and turn on lintrunner for functional_collective (#115648) 2023-12-13 11:19:29 +00:00
_functional_collectives.py Implements permute_tensor in functional collectives (#115078) 2023-12-19 18:33:28 +00:00
_state_dict_utils.py [DCP][BE] Move DCP._state_dict_utils out from DCP (#115523) 2023-12-13 08:59:48 +00:00
argparse_util.py Fix docstring errors in _composable_state.py, remote_device.py, value_ranges.py, utils.py, run.py, rendezvous.py, launch.py, argparse_util.py, __init__.py, _cycles.py (#112953) 2023-11-08 01:13:09 +00:00
c10d_logger.py Re-enable type checking for distributed_c10d.py (#115223) 2023-12-09 11:07:54 +00:00
collective_utils.py [Reland] Update mypy to 1.4.1 (#105227) 2023-07-15 20:30:20 +00:00
constants.py Switch env variable use in test harnesses to the non-deprecated names to fix warnings (#114880) 2023-12-01 20:08:23 +00:00
CONTRIBUTING.md
device_mesh.py Update device_mesh.py docs imports (#116074) 2023-12-19 09:44:55 +00:00
distributed_c10d.py Adds allreduce to inductor remap (#115950) 2023-12-18 22:00:22 +00:00
launch.py Fix docstring errors in _composable_state.py, remote_device.py, value_ranges.py, utils.py, run.py, rendezvous.py, launch.py, argparse_util.py, __init__.py, _cycles.py (#112953) 2023-11-08 01:13:09 +00:00
logging_handlers.py
remote_device.py Fix docstring errors in _composable_state.py, remote_device.py, value_ranges.py, utils.py, run.py, rendezvous.py, launch.py, argparse_util.py, __init__.py, _cycles.py (#112953) 2023-11-08 01:13:09 +00:00
rendezvous.py Fix docstring errors in _composable_state.py, remote_device.py, value_ranges.py, utils.py, run.py, rendezvous.py, launch.py, argparse_util.py, __init__.py, _cycles.py (#112953) 2023-11-08 01:13:09 +00:00
run.py [torchrun] fix incorrect warning for non static backend (#114335) 2023-11-22 20:09:14 +00:00
utils.py [fsdp][torch.compile] FSDP changes (#115497) 2023-12-19 18:44:36 +00:00