pytorch

745739d8 - reland of #80545 w skip rocm (#81738)

reland of #80545 w skip rocm (#81738) Pull Request resolved: https://github.com/pytorch/pytorch/pull/81738 Approved by: https://github.com/ezyang