| .. |
|
_composable
|
[FSDP2] Added set_reduce_scatter_divide_factor (#129286)
|
2024-07-24 12:42:35 +00:00 |
|
_shard
|
typing proxy_tensor.py (#129182)
|
2024-07-12 23:17:09 +00:00 |
|
_sharded_tensor
|
|
|
|
_sharding_spec
|
|
|
|
_spmd
|
[BE] typing for decorators - jit/_decompositions (#131566)
|
2024-07-24 20:28:28 +00:00 |
|
_symmetric_memory
|
SymmetricMemory-based, low contention intra-node all-gather and reduce-scatter (#130583)
|
2024-07-23 23:37:48 +00:00 |
|
_tensor
|
[debug][dtensor] implemented activation checkpointing differentiation (#130996)
|
2024-07-23 23:44:56 +00:00 |
|
_tools
|
[BE]: Update flake8-comprehensions and enable C420 (#130699)
|
2024-07-16 13:47:49 +00:00 |
|
algorithms
|
[BE] Make ActivationWrapper an abstract class (#129808)
|
2024-07-02 04:29:43 +00:00 |
|
autograd
|
|
|
|
benchmarks
|
[Ez][BE]: Enable new stable ruff rules (#129825)
|
2024-07-02 14:47:10 +00:00 |
|
checkpoint
|
[BE] mypy: disallow untyped decorators (#131428)
|
2024-07-23 21:50:55 +00:00 |
|
elastic
|
elastic/store: use wait instead of get for barrier (#130148)
|
2024-07-08 19:53:42 +00:00 |
|
examples
|
|
|
|
fsdp
|
[BE][Easy] fix ruff rule needless-bool (SIM103) (#130206)
|
2024-07-14 08:17:52 +00:00 |
|
launcher
|
|
|
|
nn
|
|
|
|
optim
|
[BE] mypy: disallow untyped decorators (#131428)
|
2024-07-23 21:50:55 +00:00 |
|
pipelining
|
support zb1p and zb2p algorithms (#130752)
|
2024-07-24 17:58:46 +00:00 |
|
rpc
|
[BE] mypy: disallow untyped decorators (#131428)
|
2024-07-23 21:50:55 +00:00 |
|
tensor
|
[tp] improve SequenceParallel and its documentation (#131346)
|
2024-07-23 03:57:01 +00:00 |
|
__init__.py
|
|
|
|
_checkpointable.py
|
|
|
|
_composable_state.py
|
|
|
|
_functional_collectives.py
|
|
|
|
_functional_collectives_impl.py
|
|
|
|
_state_dict_utils.py
|
[pytorch] fix the leak for pinned memory when using _create_cpu_state… (#131270)
|
2024-07-23 15:47:21 +00:00 |
|
argparse_util.py
|
|
|
|
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
|
|
|
|
constants.py
|
|
|
|
CONTRIBUTING.md
|
|
|
|
device_mesh.py
|
[DeviceMesh][Reland] Only include the real thread_id in DeviceMesh hash under threaded backend (#130495) (#130685)
|
2024-07-15 20:05:26 +00:00 |
|
distributed_c10d.py
|
[c10] add an option to pg_config split share (#130877)
|
2024-07-19 21:11:26 +00:00 |
|
launch.py
|
|
|
|
logging_handlers.py
|
|
|
|
remote_device.py
|
[BE][Easy] fix ruff rule needless-bool (SIM103) (#130206)
|
2024-07-14 08:17:52 +00:00 |
|
rendezvous.py
|
|
|
|
run.py
|
|
|
|
utils.py
|
[Reland][PT-D] Relaxed contract to allow Sequence[nn.Module] (#127773) (#130947)
|
2024-07-17 22:40:13 +00:00 |
44fdf24967