Print collective (#97544)
Prints the collective running in TDD.
Differential Revision: [D44347417](https://our.internmc.facebook.com/intern/diff/D44347417/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97544
Approved by: https://github.com/zhaojuanmao
