pytorch/torch/distributed
2022-11-19 18:01:25 +00:00
..
_composable [WIP] Composable API: replicate and DistributedState (#87649) 2022-11-17 03:06:31 +00:00
_shard [torchrec] [composable] update ShardedEmbeddingBagCollection to be use registered EBCs with shardedTensors as registered modules (#758) (#88026) 2022-11-17 04:26:13 +00:00
_sharded_tensor
_sharding_spec Add __all__ for a few distributed modules plus a little typing (reland) (#84872) 2022-09-13 21:57:49 +00:00
_spmd Remove eager mode support form CommTensor (#84978) 2022-09-14 17:23:23 +00:00
_tensor [PT-D][1/N] Sync TP Beta change to prod (#89242) 2022-11-19 18:01:25 +00:00
algorithms Fix typos used in documents under torch directory (#88300) 2022-11-02 09:38:13 +00:00
autograd Integrate xdoctest - Rebased (#82797) 2022-08-12 02:08:01 +00:00
benchmarks Fix typos in messages under torch (#89049) 2022-11-17 04:18:14 +00:00
checkpoint [PT-D][Checkpoint] Update import and update docstring for distributed checkpoint (#89256) 2022-11-18 09:49:39 +00:00
elastic Fix typos in messages under torch (#89049) 2022-11-17 04:18:14 +00:00
fsdp [FSDP] Add fast path for NO_SHARD clip_grad_norm_() (#89137) 2022-11-16 22:08:50 +00:00
launcher Add __all__ to torch.distributed and tensorboard submodules (#80444) 2022-06-28 16:33:22 +00:00
nn [nn] add remove_duplicate flag to named_parameters (#759) (#88090) 2022-11-09 00:09:20 +00:00
optim Upstream apply_optim_in_backward from TorchRec (#87397) (#88539) 2022-11-05 18:28:07 +00:00
pipeline Deprecate TypedStorage, its derived classes, and all of their public methods (#85303) 2022-11-08 18:11:01 +00:00
rpc [Python] refactor slices on sorted (#86995) 2022-10-25 04:07:19 +00:00
__init__.py Add torch.distributed.DistBackendError exception type, thrown from C10D_NCCL_CHECK (#88134) 2022-11-08 13:26:42 +00:00
argparse_util.py
c10d_error_logger.py [C10D][BE] Add exception handlers to c10d collectives function (#87643) (#87988) 2022-10-29 04:38:34 +00:00
constants.py
CONTRIBUTING.md
distributed_c10d.py [small] Update error message (#89294) 2022-11-19 00:21:14 +00:00
launch.py Integrate xdoctest - Rebased (#82797) 2022-08-12 02:08:01 +00:00
logging_handlers.py [C10D][BE] Add exception handlers to c10d collectives function (#87643) (#87988) 2022-10-29 04:38:34 +00:00
remote_device.py
rendezvous.py [rpc/distributed] eliminate code duplication in distributed/rendezvou… (#81577) 2022-07-22 16:21:00 +00:00
run.py Integrate xdoctest - Rebased (#82797) 2022-08-12 02:08:01 +00:00
utils.py [DDP] Add PackedSequence support when device_ids is specified (#86614) 2022-10-10 21:50:59 +00:00