pytorch/benchmarks/distributed
Rohan Varma 5021582fe6 Fix benchmarks/distributed/ddp/benchmark.py (#51095)
Summary:
Fixes the issue reported in https://github.com/pytorch/pytorch/issues/50679 by using built-in object-based collectives. User has verified this patch works

Test with:
RANK=0 python3 pytorch-dist-benchmark.py --world-size 2 --master-addr 127.0.0.1 --master-port 23456
RANK=1 python3 pytorch-dist-benchmark.py --world-size 2 --master-addr 127.0.0.1 --master-port 23456

Pull Request resolved: https://github.com/pytorch/pytorch/pull/51095

Reviewed By: SciPioneer

Differential Revision: D26070275

Pulled By: rohan-varma

fbshipit-source-id: 59abcaac9e395bcdd8a018bf6ba07521d94b2fdf
2021-01-29 11:10:13 -08:00
..
ddp Fix benchmarks/distributed/ddp/benchmark.py (#51095) 2021-01-29 11:10:13 -08:00
pipeline [Pipe] Refactor convert_to_balance under non-test package. (#50860) 2021-01-28 12:10:21 -08:00
rpc/rl Pytorch Distributed RPC Reinforcement Learning Benchmark (Throughput and Latency) (#46901) 2021-01-11 19:02:36 -08:00