pytorch/torch/distributed
Aaron Gokaslan 29cc293725 [BE]: FURB142 - Remove set mutations. Use set update (#124551)
Uses set mutation methods instead of manually reimplementing (update, set_difference etc).

Pull Request resolved: https://github.com/pytorch/pytorch/pull/124551
Approved by: https://github.com/ezyang
2024-04-21 14:12:33 +00:00
..
_composable [FSDP2][ez] Removed error check for swap tensors flag (#124513) 2024-04-20 00:46:36 +00:00
_shard [PT] [ST] fix test_sharded_tensor (#124103) 2024-04-16 21:18:22 +00:00
_sharded_tensor
_sharding_spec
_spmd [BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261) 2024-04-17 19:29:34 +00:00
_tensor [BE]: FURB142 - Remove set mutations. Use set update (#124551) 2024-04-21 14:12:33 +00:00
_tools
algorithms
autograd
benchmarks
checkpoint [BE]: FURB142 - Remove set mutations. Use set update (#124551) 2024-04-21 14:12:33 +00:00
elastic [BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261) 2024-04-17 19:29:34 +00:00
examples
fsdp [PT] [FSDP] fix HSDP sharding placement (#123778) 2024-04-12 00:05:49 +00:00
launcher
nn
optim
pipeline Add missing interfaces of torch.optim.swa_utils (#117036) 2024-04-12 17:17:36 +00:00
rpc
tensor
__init__.py [C10D] Add dist.get_node_local_rank helper (#123992) 2024-04-16 00:09:46 +00:00
_composable_state.py
_functional_collectives_impl.py [functional collective] change the Python APIs to only use the native funcol ops (#123777) 2024-04-13 03:08:36 +00:00
_functional_collectives.py all_gather and reduce_scatter autograd (#123989) 2024-04-17 21:32:22 +00:00
_state_dict_utils.py [BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261) 2024-04-17 19:29:34 +00:00
argparse_util.py
c10d_logger.py [DCP] Adds better handling in logging of specific kwargs (#123658) 2024-04-11 21:09:38 +00:00
collective_utils.py
constants.py
CONTRIBUTING.md
device_mesh.py [DeviceMesh] Make dtype of mesh tensor from init_device_mesh() consistent with directly calling DeviceMesh() (#123677) 2024-04-10 09:14:34 +00:00
distributed_c10d.py [BE]: Update ruff to 0.4.1 (#124549) 2024-04-21 14:06:23 +00:00
launch.py [Docs][Distributed] Add migration notes for --local-rank option style change for torchrun in PyTorch 2.0 (#109480) 2024-04-16 05:51:57 +00:00
logging_handlers.py
remote_device.py
rendezvous.py
run.py [Docs][Distributed] Add migration notes for --local-rank option style change for torchrun in PyTorch 2.0 (#109480) 2024-04-16 05:51:57 +00:00
utils.py