pytorch

b168c4fa - Make CommTensor Generic to arguments and outputs structures (#84576)

Make CommTensor Generic to arguments and outputs structures (#84576) Pull Request resolved: https://github.com/pytorch/pytorch/pull/84576 Approved by: https://github.com/aazzolini