[ROCm][CI] Route unittests to MI350 ecosystem runners - #9567
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/9567
Note: Links to docs will display an error until the docs builds have been completed. This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
The following ciflow label(s) have been added but CI has not been triggered yet because the workflows are awaiting approval:
Once a maintainer approves the workflows (scroll to the bottom of the PR page), the corresponding CI jobs will be triggered automatically. Please ping one of the reviewers if you do not have access to approve and run workflows. |
amdfaa
changed the title
amdfaa
changed the title
Switch the unittests-rocm job in .github/workflows/tests.yml to the MI350 (gfx950) ecosystem runner label linux.rocm.gpu.ecosystem.gfx950.1, used by downstream PyTorch ecosystem projects. Made with Cursor
f41e543 to
70575a6
Compare
amdfaa
changed the title
| branches: | ||
| - nightly | ||
| - main | ||
| - release/* |
There was a problem hiding this comment.
| - release/* | |
| tags: | |
| - ciflow/rocm/* |
Let's also add the tag push trigger so PRs can run ROCm CI
|
Thanks for the PR - I won't merge it because I'm in the process of moving the ROCJPEG decoder into TorchCodec (and I'll remove it from TorchVision) in meta-pytorch/torchcodec#1554. Since this job is just testing the ROCJPEG decoder, it'll soon be removed. But I'll follow-up in TorchCodec with the runner update. Thanks! |
|
@NicolasHug Thanks for the update |
3 participants
Summary
Switch the ROCm
unittests-rocmjob in.github/workflows/tests.ymlto theshared MI350 (gfx950) ecosystem runner scale set used by downstream PyTorch
ecosystem projects (e.g.
pytorch/helion,pytorch/torchtitan):linux.rocm.gpu.ecosystem.gfx950.1Test plan
unittests-rocmjob should schedule and run on alinux.rocm.gpu.ecosystem.gfx950.1runner (same-repo PR / push /workflow_dispatch; the job is skipped for fork PRs by design because fork PRs
don't receive a GitHub OIDC id-token for ECR/AWS auth).
Made with Cursor
cc @jeffdaily @jithunnair-amd