| .. |
|
_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
|
Replace enforce_dtype with dtype in ShardedTensor.gather (#110561)
|
2023-10-05 23:16:23 +00:00 |
|
_sharded_tensor
|
|
|
|
_sharding_spec
|
|
|
|
_spmd
|
Fix typo under torch directory (#110824)
|
2023-10-09 19:16:43 +00:00 |
|
_tensor
|
device agnostic: torch.cpu.set_device (#110716)
|
2023-10-09 23:00:15 +00:00 |
|
_tools
|
|
|
|
algorithms
|
fix: flake8-bugbear code B024 (#107265)
|
2023-10-04 23:52:52 +00:00 |
|
autograd
|
|
|
|
benchmarks
|
[BE] Enable ruff's UP rules and autoformat distributed/ (#105433)
|
2023-07-19 14:27:11 +00:00 |
|
checkpoint
|
Add current_device() to torch.cpu (#110987)
|
2023-10-11 05:13:10 +00:00 |
|
elastic
|
Format: fixing multiple string concatenation in single line (#106013)
|
2023-07-26 18:39:18 +00:00 |
|
examples
|
|
|
|
fsdp
|
[FSDP] Change _create_chunk_dtensor in fsdp/_shard_utils.py to use public API from DTensor (#110831)
|
2023-10-10 21:04:27 +00:00 |
|
launcher
|
|
|
|
nn
|
[BE]: Update ruff to 0.285 (#107519)
|
2023-08-22 23:16:38 +00:00 |
|
optim
|
[Distributed] Small nits to apply_optimizer_in_backward (#110903)
|
2023-10-11 07:45:45 +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
|
Allow public access for imports (#108914)
|
2023-09-28 06:05:59 +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
|
Add functional collective all_to_all_single and support it in Inductor (#110195)
|
2023-10-05 23:11:51 +00:00 |
|
_functional_collectives.py
|
switch dtensor and functional collective to use optree (#110670)
|
2023-10-08 03:05:39 +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
|
Revert "Reland "[C10] PG observability hooks. (#108815)" (#110907)"
|
2023-10-11 00:23:42 +00:00 |
|
launch.py
|
|
|
|
logging_handlers.py
|
|
|
|
remote_device.py
|
|
|
|
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
|
[Easy] Remove unused return type from utils (#110887)
|
2023-10-10 09:02:11 +00:00 |
de370eb313