pytorch/torch/distributed
Sean Lawlor 34c9f5a8da [DDP Communication Hook] Update get_tensor and set_tensor to be cleaner naming conventions (buffer() and set_buffer()) (#62662)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62662

Replaced the methods set_tensor(.) and get_tensor() in the python exposed API from the C++ logic with buffer() and set_buffer(.) to be a cleaner interface.

Reviewed By: SciPioneer

Differential Revision: D30012869

fbshipit-source-id: bd8efab583dd89c96f9aeb3dd48a12073f0b1482
2021-08-04 09:27:31 -07:00
..
_sharded_tensor Initialize RRefs only when explicitly asked for. (#62618) 2021-08-02 22:17:17 -07:00
_sharding_spec [sharded_tensor] add new init_from_local_shards API (#60479) 2021-07-29 22:04:13 -07:00
algorithms [DDP Communication Hook] Update get_tensor and set_tensor to be cleaner naming conventions (buffer() and set_buffer()) (#62662) 2021-08-04 09:27:31 -07:00
autograd
benchmarks
elastic Add logging for dynamic rendezvous (#61822) 2021-07-26 09:39:09 -07:00
launcher [torchelastic] Make sure rdzv_configs[timeout] is not getting overwritten (#61471) 2021-07-09 15:27:00 -07:00
nn
optim Make _Join, _Joinable, _JoinHook public (#62605) 2021-08-03 12:20:11 -07:00
pipeline Adds _LazyInstanceNorm and LazyInstanceNormXd (#60982) 2021-07-21 06:45:45 -07:00
rpc [profiler][refactor] Refactor the usage of legacy profiler implementation (#61931) 2021-08-03 18:51:29 -07:00
__init__.py
argparse_util.py
constants.py
CONTRIBUTING.md
distributed_c10d.py Adding warning on isend about modifying after send (#61875) 2021-07-29 07:37:18 -07:00
launch.py [torch] Various improvements to torch.distributed.launch and torch.distributed.run (#61294) 2021-07-08 16:28:06 -07:00
rendezvous.py [torch/elastic] Fix the agent store key prefix used by workers (#61590) 2021-07-13 14:57:27 -07:00
run.py [torch] Set nproc_per_node to 1 (#61552) 2021-07-13 13:35:25 -07:00
utils.py