@@ -88,18 +88,23 @@ jobs:
88
88
cuda-arch-list : ' 8.0'
89
89
test-matrix : |
90
90
{ include: [
91
- { config: "inductor_huggingface_perf", shard: 1, num_shards: 3, runner: "linux.aws.a100" },
92
- { config: "inductor_huggingface_perf", shard: 2, num_shards: 3, runner: "linux.aws.a100" },
93
- { config: "inductor_huggingface_perf", shard: 3, num_shards: 3, runner: "linux.aws.a100" },
94
- { config: "inductor_timm_perf", shard: 1, num_shards: 5, runner: "linux.aws.a100" },
95
- { config: "inductor_timm_perf", shard: 2, num_shards: 5, runner: "linux.aws.a100" },
96
- { config: "inductor_timm_perf", shard: 3, num_shards: 5, runner: "linux.aws.a100" },
97
- { config: "inductor_timm_perf", shard: 4, num_shards: 5, runner: "linux.aws.a100" },
98
- { config: "inductor_timm_perf", shard: 5, num_shards: 5, runner: "linux.aws.a100" },
99
- { config: "inductor_torchbench_perf", shard: 1, num_shards: 4, runner: "linux.aws.a100" },
100
- { config: "inductor_torchbench_perf", shard: 2, num_shards: 4, runner: "linux.aws.a100" },
101
- { config: "inductor_torchbench_perf", shard: 3, num_shards: 4, runner: "linux.aws.a100" },
102
- { config: "inductor_torchbench_perf", shard: 4, num_shards: 4, runner: "linux.aws.a100" },
91
+ { config: "inductor_huggingface_perf", shard: 1, num_shards: 5, runner: "linux.aws.a100" },
92
+ { config: "inductor_huggingface_perf", shard: 2, num_shards: 5, runner: "linux.aws.a100" },
93
+ { config: "inductor_huggingface_perf", shard: 3, num_shards: 5, runner: "linux.aws.a100" },
94
+ { config: "inductor_huggingface_perf", shard: 4, num_shards: 5, runner: "linux.aws.a100" },
95
+ { config: "inductor_huggingface_perf", shard: 5, num_shards: 5, runner: "linux.aws.a100" },
96
+ { config: "inductor_timm_perf", shard: 1, num_shards: 6, runner: "linux.aws.a100" },
97
+ { config: "inductor_timm_perf", shard: 2, num_shards: 6, runner: "linux.aws.a100" },
98
+ { config: "inductor_timm_perf", shard: 3, num_shards: 6, runner: "linux.aws.a100" },
99
+ { config: "inductor_timm_perf", shard: 4, num_shards: 6, runner: "linux.aws.a100" },
100
+ { config: "inductor_timm_perf", shard: 5, num_shards: 6, runner: "linux.aws.a100" },
101
+ { config: "inductor_timm_perf", shard: 6, num_shards: 6, runner: "linux.aws.a100" },
102
+ { config: "inductor_torchbench_perf", shard: 1, num_shards: 6, runner: "linux.aws.a100" },
103
+ { config: "inductor_torchbench_perf", shard: 2, num_shards: 6, runner: "linux.aws.a100" },
104
+ { config: "inductor_torchbench_perf", shard: 3, num_shards: 6, runner: "linux.aws.a100" },
105
+ { config: "inductor_torchbench_perf", shard: 4, num_shards: 6, runner: "linux.aws.a100" },
106
+ { config: "inductor_torchbench_perf", shard: 5, num_shards: 6, runner: "linux.aws.a100" },
107
+ { config: "inductor_torchbench_perf", shard: 6, num_shards: 6, runner: "linux.aws.a100" },
103
108
]}
104
109
selected-test-configs : ${{ inputs.benchmark_configs }}
105
110
secrets : inherit
0 commit comments