saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/torch/distributed
History
Yifu Wang 93a4671746 Add out_dtypes to fused_all_gather_scaled_matmul's args (#131831) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/131831
Approved by: https://github.com/weifengpy
ghstack dependencies: #131410
2024-07-27 11:07:43 +00:00
..
_composable [Traceable FSDP2] Add partial-graph (graph-break) unit tests (#131747) 2024-07-26 02:51:57 +00:00
_shard typing proxy_tensor.py (#129182) 2024-07-12 23:17:09 +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
_spmd [BE] typing for decorators - distributed/_tensor/ops/utils (#131571) 2024-07-25 22:24:19 +00:00
_symmetric_memory Add out_dtypes to fused_all_gather_scaled_matmul's args (#131831) 2024-07-27 11:07:43 +00:00
_tensor [BE] typing for decorators - distributed/_tensor/ops/utils (#131571) 2024-07-25 22:24:19 +00:00
_tools [BE]: Update flake8-comprehensions and enable C420 (#130699) 2024-07-16 13:47:49 +00:00
algorithms [BE] [Reland] Make nn.Module state_dict load_state_dict pre-hook and state_dict post-hook public (#131690) 2024-07-26 18:14:07 +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 [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 [BE][Easy] enable UFMT for torch/distributed/ (#128870) 2024-06-22 18:53:28 +00:00
fsdp Added hpu backend support in fsdp utils (#127757) 2024-07-27 03:30:59 +00:00
launcher [BE][Easy] enable UFMT for torch/distributed/ (#128870) 2024-06-22 18:53:28 +00:00
nn [BE][Easy] enable UFMT for torch/distributed/ (#128870) 2024-06-22 18:53:28 +00:00
optim [BE] typing for decorators - _jit_internal (#131573) 2024-07-25 22:24:19 +00:00
pipelining Fix public API tests (#131386) 2024-07-26 23:38:43 +00:00
rpc [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
tensor [dtensor] move ops to private (#131211) 2024-07-25 20:59:55 +00:00
__init__.py [BE][Easy] enable UFMT for torch/distributed/ (#128870) 2024-06-22 18:53:28 +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.py Not remove collective ops in dce since they have side-effect (#131023) 2024-07-26 03:03:32 +00:00
_functional_collectives_impl.py [BE][Easy] enable UFMT for torch/distributed/ (#128870) 2024-06-22 18:53:28 +00:00
_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 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] Remove _parent_mesh as an attribute from DeviceMesh and remove it from DeviceMesh's hash (#131636) 2024-07-25 22:47:22 +00:00
distributed_c10d.py [c10d] Fix split_group usage when there is a single rank (#131824) 2024-07-26 18:11:17 +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