remove thunk fix now that ROCm CI images are >= ROCm 3.5 (#43226)
Summary:
Also, relax BUILD_ENVIRONMENT exact match to rocm when installing pip packages for tests.
CC ezyang xw285cornell sunway513
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43226
Reviewed By: colesbury
Differential Revision: D23200460
Pulled By: xw285cornell
fbshipit-source-id: 11cd889cc320d0249d7ebea4da261bfe779e82ac