Write OpInfo for dist (#55092)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/53516
cc anjali411
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55092
Reviewed By: nikithamalgifb
Differential Revision: D27493577
Pulled By: anjali411
fbshipit-source-id: c7e8400a20bbc7138249b249e322b3b23e112336
