pytorch

e549e974 - Upgrade CI to ROCm 5.1

Upgrade CI to ROCm 5.1 Pull Request resolved: https://github.com/pytorch/pytorch/pull/76365 Approved by: https://github.com/seemethere, https://github.com/jeffdaily