ROCm thunk work-around for future transition to ROCm 3.5.1 (#40181)
Summary:
ROCm CI hosts will have their kernels upgraded first to ROCm 3.5.1. CI images will follow soon after. Due to the thunk/kernel mismatch during the interim, this PR will detect the mismatch and upgrade the thunk during the build. This PR will be reverted once migration to ROCm 3.5.1 images is complete.
CC ezyang xw285cornell
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40181
Differential Revision: D22104488
Pulled By: xw285cornell
fbshipit-source-id: 7192e1d0bb25bfb814e9a85efb4aa29d0e52b460