Split PyTorch ROCm tests as 2 CI jobs to run in parallel (#26380)
Summary:
ROCm CI jobs are running on Jenkins. They have the "-test{1,2}" parts in "JOB_BASE_NAME", not "BUILD_ENVIRONMENT".
Pull Request resolved: https://github.com/pytorch/pytorch/pull/26380
Differential Revision: D17439523
Pulled By: bddppq
fbshipit-source-id: 31e2a986d1b7ea40c90ab399a3c1e0a328ae3a92