Use ROCm MI325 runners for trunk.yml (#161184)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/161184
Approved by: https://github.com/jeffdaily
This commit is contained in:
Jithun Nair 2025-08-22 16:18:55 +00:00 committed by PyTorch MergeBot
parent c7a77470c5
commit 7fcdd8d6af

View File

@ -201,9 +201,9 @@ jobs:
sync-tag: rocm-build
test-matrix: |
{ include: [
{ config: "default", shard: 1, num_shards: 2, runner: "linux.rocm.gpu.2" },
{ config: "default", shard: 2, num_shards: 2, runner: "linux.rocm.gpu.2" },
{ config: "distributed", shard: 1, num_shards: 1, runner: "linux.rocm.gpu.4" },
{ config: "default", shard: 1, num_shards: 2, runner: "linux.rocm.gpu.gfx942.1" },
{ config: "default", shard: 2, num_shards: 2, runner: "linux.rocm.gpu.gfx942.1" },
{ config: "distributed", shard: 1, num_shards: 1, runner: "linux.rocm.gpu.gfx942.4" },
]}
secrets: inherit