[ROCm] use ncclAllToAll for rocm
use ncclAllToAll for rocm version > 5.0; https://github.com/ROCmSoftwarePlatform/rccl/pull/503
detail on ncclAllToAll:
https://github.com/ROCmSoftwarePlatform/rccl/pull/503
@jithunnair-amd @amathews-amd
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75128
Approved by: https://github.com/wenkaidu, https://github.com/yzygitzh, https://github.com/seemethere
