pytorch/torch/distributed
2023-09-26 22:24:50 +00:00
..
_composable [Dynamo x FSDP][2/x] Small changes to distributed to make it dynamo friendly (#106886) 2023-08-11 22:35:50 +00:00
_shard Update chunk_sharding_spec.py (#108915) 2023-09-15 21:43:15 +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 [dtensor][3/n] use dedicated TensorMeta instead of the fx one (#108261) 2023-09-13 04:08:02 +00:00
_tensor [dtensor][7/n] remove reduction rule (#109144) 2023-09-26 22:24:50 +00:00
_tools Fix typos under torch/distributed directory (#95638) 2023-03-27 21:13:44 +00:00
algorithms cleanup typos; redundant parentheses (#109003) 2023-09-11 17:09:17 +00:00
autograd [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
benchmarks [BE] Enable ruff's UP rules and autoformat distributed/ (#105433) 2023-07-19 14:27:11 +00:00
checkpoint Fix invalid arg to getLogger in torch distributed checkpoint (#110008) 2023-09-25 18:21:18 +00:00
elastic Format: fixing multiple string concatenation in single line (#106013) 2023-07-26 18:39:18 +00:00
examples Fix typos under torch/distributed directory (#95638) 2023-03-27 21:13:44 +00:00
fsdp [3/N][2D] Enable training with new 2D flow (#110034) 2023-09-26 09:14:15 +00:00
launcher Convert logging f-strings to use % format, part four (#98705) 2023-04-11 13:17:59 +00:00
nn [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
optim Improve type annotations for jit.script (#108782) 2023-09-13 19:20:25 +00:00
pipeline [BE]: Update ruff to 0.285 (#107519) 2023-08-22 23:16:38 +00:00
rpc [BE] Enable ruff's UP rules and autoformat distributed/ (#105433) 2023-07-19 14:27:11 +00:00
tensor [3/N][2D] Enable training with new 2D flow (#110034) 2023-09-26 09:14:15 +00:00
__init__.py [RESUBMIT] Standardize on error types for distributed errors. (#108191) 2023-08-30 21:47:39 +00:00
_composable_state.py [Dynamo x FSDP][2/x] Small changes to distributed to make it dynamo friendly (#106886) 2023-08-11 22:35:50 +00:00
_functional_collectives_impl.py [functional collectives] Move back to registering finalizers on wrappers. (#107250) 2023-08-17 21:08:28 +00:00
_functional_collectives.py Add support for tolist on AsyncCollectiveTensor (#109377) 2023-09-15 21:48:13 +00:00
argparse_util.py
c10d_logger.py [C10D][Logger]Add more info to c10d logger (#107331) 2023-08-28 15:10:56 +00:00
collective_utils.py [Reland] Update mypy to 1.4.1 (#105227) 2023-07-15 20:30:20 +00:00
constants.py
CONTRIBUTING.md
distributed_c10d.py [C10D] Even more store scalability work. (#109218) 2023-09-22 21:27:09 +00:00
launch.py Fix typos under torch/distributed directory (#95638) 2023-03-27 21:13:44 +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 [BE]: pyupgrade Python to 3.8 - imports and object inheritance only (#94308) 2023-02-07 21:10:56 +00:00
rendezvous.py [c10d] Add TCPSTore libuv backend support to c10d rendezvous. (#108284) 2023-09-07 21:39:58 +00:00
run.py [RFC] Change --standalone to bind to a random port (#107734) 2023-08-25 22:13:44 +00:00
utils.py Enable custom device support in fsdp checkpoint (#107289) 2023-08-25 11:50:03 +00:00