Add support for Rocm Runners to linux_job

[atalman]

Make sure Rocm runners can execute:
https://github.com/pytorch/test-infra/blob/main/.github/workflows/linux_job.yml

So that we can execute rocm CI test on domain libraries:
Example PR: pytorch/ao#999

cc @jithunnair-amd