[ROCm] tests set PYTORCH_TESTING_DEVICE_ONLY_FOR="cuda"
This env var works similarly for both cuda and rocm builds. Consider removing the ROCm-specific change in #55069 which came before the addition of this env var for cuda.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76668
Approved by: https://github.com/janeyx99