[8/N] [Dispatchable Collectives] Update allgather with CPU / CUDA implementations (#84423)
### Changes
- Updates for the allgather collective
### Context
https://github.com/pytorch/pytorch/issues/86225
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84423
Approved by: https://github.com/kwen2501