[ROCm] Enable wrongly skipped tests on CPU on ROCm (#36968)
Summary:
`skipIfRocm` skips the test on ROCm regardless of device type [CPU or GPU]. `skipCUDAIfRocm` skips only on GPU on ROCm and runs the test on CPU.
ezyang iotamudelta
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36968
Differential Revision: D21149721
Pulled By: ezyang
fbshipit-source-id: 361811b0b307f17193ad72ee8bcc7f2c65ce6203