| .. |
|
_composable
|
Add early_stop kwarg to torch.utils.checkpoint (#160781)
|
2025-08-26 22:32:35 +00:00 |
|
_pycute
|
[CuTe] Change the logic of pycute manipulation ops like coalesce, complement from co-lex to lex (#162690)
|
2025-09-16 19:53:45 +00:00 |
|
_shard
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
_sharded_tensor
|
|
|
|
_sharding_spec
|
|
|
|
_symmetric_memory
|
[Reland][163423] Promote @requires_nvshmem instead of enable_triton (#163549)
|
2025-09-25 23:02:00 +00:00 |
|
_tensor
|
|
|
|
_tools
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
algorithms
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
autograd
|
[remove untyped defs] batch 1 (#157011)
|
2025-06-30 23:54:40 +00:00 |
|
benchmarks
|
|
|
|
checkpoint
|
[DCP] DTensor slice dequantization with proper block alignment (#163532)
|
2025-09-23 16:48:16 +00:00 |
|
elastic
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
examples
|
|
|
|
fsdp
|
[FSDP2] idempotent reset_sharded_param: no-op if _local_tensor is already padded (#163130)
|
2025-09-18 09:20:37 +00:00 |
|
launcher
|
154849 Add support to handle IGUSR1 and SIGUSR2 in multiprocessing (#160690)
|
2025-09-09 22:23:06 +00:00 |
|
nn
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
optim
|
[BE][5/16] fix typos in torch/ (torch/distributed/) (#156315)
|
2025-06-23 02:57:28 +00:00 |
|
pipelining
|
Fix pipeline parallelism not correctly initializing backwards stages when evaluating before training. (#162823)
|
2025-09-25 15:13:19 +00:00 |
|
rpc
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
tensor
|
[distributed] Remove python code older than 3.10 (#163613)
|
2025-09-28 04:15:24 +00:00 |
|
__init__.py
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
_checkpointable.py
|
|
|
|
_composable_state.py
|
|
|
|
_dist2.py
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
_functional_collectives_impl.py
|
|
|
|
_functional_collectives.py
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
_mesh_layout.py
|
[CuTe] Add layout overlap checking util function in _MeshLayout (#163367)
|
2025-09-27 00:22:14 +00:00 |
|
_serialization.py
|
[BE][5/16] fix typos in torch/ (torch/distributed/) (#156315)
|
2025-06-23 02:57:28 +00:00 |
|
_state_dict_utils.py
|
fix-unpin-memory-tensor-param (#160992)
|
2025-08-26 21:55:25 +00:00 |
|
argparse_util.py
|
|
|
|
c10d_logger.py
|
|
|
|
collective_utils.py
|
[C10D] add _summarize_ranks util (#160284)
|
2025-08-28 00:17:53 +00:00 |
|
constants.py
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
CONTRIBUTING.md
|
fix torch/distributed contributing doc (#158934)
|
2025-07-28 17:01:05 +00:00 |
|
device_mesh.py
|
[DeviceMesh][ez] Extract the pg creation as a util function (#163930)
|
2025-09-26 20:42:58 +00:00 |
|
distributed_c10d.py
|
Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)"
|
2025-09-25 13:47:46 +00:00 |
|
launch.py
|
|
|
|
logging_handlers.py
|
|
|
|
remote_device.py
|
|
|
|
rendezvous.py
|
[BE][5/16] fix typos in torch/ (torch/distributed/) (#156315)
|
2025-06-23 02:57:28 +00:00 |
|
run.py
|
[distributed] Remove python code older than 3.10 (#163613)
|
2025-09-28 04:15:24 +00:00 |
|
utils.py
|
|
|