Skip to content

Commit 59b6694

Browse files
jeanschmidthuydhn
andauthored
Migrate inductor-perf-test-nightly.yml to use linux.aws.a100 (pytorch#139657)
Co-authored-by: Huy Do <[email protected]>
1 parent 6734cb7 commit 59b6694

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

.github/workflows/inductor-perf-test-nightly.yml

+12-12
Original file line numberDiff line numberDiff line change
@@ -87,18 +87,18 @@ jobs:
8787
cuda-arch-list: '8.0'
8888
test-matrix: |
8989
{ include: [
90-
{ config: "inductor_huggingface_perf", shard: 1, num_shards: 3, runner: "linux.gcp.a100.large" },
91-
{ config: "inductor_huggingface_perf", shard: 2, num_shards: 3, runner: "linux.gcp.a100.large" },
92-
{ config: "inductor_huggingface_perf", shard: 3, num_shards: 3, runner: "linux.gcp.a100.large" },
93-
{ config: "inductor_timm_perf", shard: 1, num_shards: 5, runner: "linux.gcp.a100.large" },
94-
{ config: "inductor_timm_perf", shard: 2, num_shards: 5, runner: "linux.gcp.a100.large" },
95-
{ config: "inductor_timm_perf", shard: 3, num_shards: 5, runner: "linux.gcp.a100.large" },
96-
{ config: "inductor_timm_perf", shard: 4, num_shards: 5, runner: "linux.gcp.a100.large" },
97-
{ config: "inductor_timm_perf", shard: 5, num_shards: 5, runner: "linux.gcp.a100.large" },
98-
{ config: "inductor_torchbench_perf", shard: 1, num_shards: 4, runner: "linux.gcp.a100.large" },
99-
{ config: "inductor_torchbench_perf", shard: 2, num_shards: 4, runner: "linux.gcp.a100.large" },
100-
{ config: "inductor_torchbench_perf", shard: 3, num_shards: 4, runner: "linux.gcp.a100.large" },
101-
{ config: "inductor_torchbench_perf", shard: 4, num_shards: 4, runner: "linux.gcp.a100.large" },
90+
{ config: "inductor_huggingface_perf", shard: 1, num_shards: 3, runner: "linux.aws.a100" },
91+
{ config: "inductor_huggingface_perf", shard: 2, num_shards: 3, runner: "linux.aws.a100" },
92+
{ config: "inductor_huggingface_perf", shard: 3, num_shards: 3, runner: "linux.aws.a100" },
93+
{ config: "inductor_timm_perf", shard: 1, num_shards: 5, runner: "linux.aws.a100" },
94+
{ config: "inductor_timm_perf", shard: 2, num_shards: 5, runner: "linux.aws.a100" },
95+
{ config: "inductor_timm_perf", shard: 3, num_shards: 5, runner: "linux.aws.a100" },
96+
{ config: "inductor_timm_perf", shard: 4, num_shards: 5, runner: "linux.aws.a100" },
97+
{ config: "inductor_timm_perf", shard: 5, num_shards: 5, runner: "linux.aws.a100" },
98+
{ config: "inductor_torchbench_perf", shard: 1, num_shards: 4, runner: "linux.aws.a100" },
99+
{ config: "inductor_torchbench_perf", shard: 2, num_shards: 4, runner: "linux.aws.a100" },
100+
{ config: "inductor_torchbench_perf", shard: 3, num_shards: 4, runner: "linux.aws.a100" },
101+
{ config: "inductor_torchbench_perf", shard: 4, num_shards: 4, runner: "linux.aws.a100" },
102102
]}
103103
selected-test-configs: ${{ inputs.benchmark_configs }}
104104
secrets:

0 commit comments

Comments
 (0)