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