pytorch

91378544 - fix rocm build

fix rocm build Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15945 Differential Revision: D13630505 Pulled By: zdevito fbshipit-source-id: a4d2ae1370ab475fc1711027c0c9d2a9192be195