Enable capturing of comm collective parameters (#98) (#85368)
Summary:
X-link: https://github.com/facebookresearch/torch_ucc/pull/98
Add tensor input, output, and other metadata for PyTorch comms.
Test Plan: P517138779
Reviewed By: Pavani-Panakanti
Differential Revision: D38357077
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85368
Approved by: https://github.com/H-Huang