@@ -87,18 +87,18 @@ jobs:
87
87
cuda-arch-list : ' 8.0'
88
88
test-matrix : |
89
89
{ 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" },
102
102
]}
103
103
selected-test-configs : ${{ inputs.benchmark_configs }}
104
104
secrets :
0 commit comments