| .. |
|
_composable
|
[2nd try][Traceable FSDP2] Allow tracing through FSDP2 impl in trace_rules.py (#134539)
|
2024-08-29 06:28:16 +00:00 |
|
_shard
|
[BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200)
|
2024-08-15 15:50:19 +00:00 |
|
_sharded_tensor
|
[BE][Easy] enable UFMT for torch/distributed/ (#128870)
|
2024-06-22 18:53:28 +00:00 |
|
_sharding_spec
|
[BE][Easy] enable UFMT for torch/distributed/ (#128870)
|
2024-06-22 18:53:28 +00:00 |
|
_symmetric_memory
|
[micro_pipeline_tp] support all _scaled_mm args (#131984)
|
2024-08-05 21:44:37 +00:00 |
|
_tensor
|
Revert "[reland][dtensor][MTPG] make sharding prop lru cache not shared among threads (#134509)"
|
2024-08-29 16:39:19 +00:00 |
|
_tools
|
Runtime Estimator for estimating GPU compute time (#134243)
|
2024-08-28 20:06:54 +00:00 |
|
algorithms
|
[BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200)
|
2024-08-15 15:50:19 +00:00 |
|
autograd
|
[BE][Easy] enable UFMT for torch/distributed/{algorithms,autograd,benchmarks,checkpoint,elastic}/ (#128866)
|
2024-06-18 13:51:53 +00:00 |
|
benchmarks
|
[BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200)
|
2024-08-15 15:50:19 +00:00 |
|
checkpoint
|
[DCP] Fixes the BC issue where the traversal doesn't support versions before 2.4 (#134158)
|
2024-08-28 16:31:44 +00:00 |
|
elastic
|
Revert "Adding entry-point based support for out-of-tree rendezvous plugins (#132633)"
|
2024-08-28 15:49:18 +00:00 |
|
examples
|
[BE][Easy] enable UFMT for torch/distributed/ (#128870)
|
2024-06-22 18:53:28 +00:00 |
|
fsdp
|
[FSDP] Made clip_grad_norm_ norm compute order deterministic (#134673)
|
2024-08-28 18:44:11 +00:00 |
|
launcher
|
[BE][Easy] enable UFMT for torch/distributed/ (#128870)
|
2024-06-22 18:53:28 +00:00 |
|
nn
|
Revert "added persistent option to buffers and namedbuffers (#132994)"
|
2024-08-09 18:14:53 +00:00 |
|
optim
|
Revert "[BE] typing for decorators - _jit_internal (#131573)"
|
2024-07-28 03:29:32 +00:00 |
|
pipelining
|
[BE] typing for decorators - fx/_compatibility (part 1) (#134202)
|
2024-08-22 17:07:33 +00:00 |
|
rpc
|
[BE][Easy] enable ruff rule PIE790: unnecessary pass statement (#133200)
|
2024-08-15 15:50:19 +00:00 |
|
tensor
|
Revert "[dtensor] move DTensor to public namespace (#133113)"
|
2024-08-19 05:00:19 +00:00 |
|
__init__.py
|
Remove ProcessGroupRoundRobin (#132888)
|
2024-08-08 01:07:40 +00:00 |
|
_checkpointable.py
|
[torchrec][pt-d][model store] introduce LocalShardsWrapper for DTensor (#129150)
|
2024-06-21 01:58:51 +00:00 |
|
_composable_state.py
|
|
|
|
_functional_collectives_impl.py
|
[BE][Easy] enable UFMT for torch/distributed/ (#128870)
|
2024-06-22 18:53:28 +00:00 |
|
_functional_collectives.py
|
Revert "[dtensor] move DTensor to public namespace (#133113)"
|
2024-08-19 05:00:19 +00:00 |
|
_state_dict_utils.py
|
Memory optimization for DSD for TorchTune LoRA (#134025)
|
2024-08-26 17:24:25 +00:00 |
|
argparse_util.py
|
Flip default value for mypy disallow_untyped_defs [5/11] (#127842)
|
2024-06-08 18:49:18 +00:00 |
|
c10d_logger.py
|
[DCP] Fix duplicated logging messages when enable both c10d and dcp l… (#130423)
|
2024-07-11 13:43:39 +00:00 |
|
collective_utils.py
|
[BE][Easy] enable UFMT for torch/distributed/ (#128870)
|
2024-06-22 18:53:28 +00:00 |
|
constants.py
|
[BE][Easy] enable UFMT for torch/distributed/ (#128870)
|
2024-06-22 18:53:28 +00:00 |
|
CONTRIBUTING.md
|
Clean up distributed/CONTRIBUTING.md (#128450)
|
2024-06-22 02:41:22 +00:00 |
|
device_mesh.py
|
[DeviceMesh] Add get_all_submeshes in _MeshEnv (#134275)
|
2024-08-27 14:51:19 +00:00 |
|
distributed_c10d.py
|
[CP] Rewrite ring attention backward algorithm and enablement APIs (#131351)
|
2024-08-15 16:41:51 +00:00 |
|
launch.py
|
Flip default value for mypy disallow_untyped_defs [6/11] (#127843)
|
2024-06-08 18:49:29 +00:00 |
|
logging_handlers.py
|
[BE][Easy] enable UFMT for torch/distributed/ (#128870)
|
2024-06-22 18:53:28 +00:00 |
|
remote_device.py
|
[BE][Easy] fix ruff rule needless-bool (SIM103) (#130206)
|
2024-07-14 08:17:52 +00:00 |
|
rendezvous.py
|
[BE][Easy] enable UFMT for torch/distributed/ (#128870)
|
2024-06-22 18:53:28 +00:00 |
|
run.py
|
fix torchrun log message (#131652)
|
2024-07-25 14:50:10 +00:00 |
|
utils.py
|
[Reland][PT-D] Relaxed contract to allow Sequence[nn.Module] (#127773) (#130947)
|
2024-07-17 22:40:13 +00:00 |
ab646cd805