Skip to content

Commit 5d01a28

Browse files
huydhnpytorchmergebot
authored andcommitted
Increase the number of perf benchmark shards (pytorch#145534)
Per the discussion on pytorch#140332 (comment), this adds 2 more shards for HF, 2 more for TorchBench, and 1 more for TIMM. Pull Request resolved: pytorch#145534 Approved by: https://github.com/jeanschmidt
1 parent 639dd54 commit 5d01a28

File tree

1 file changed

+17
-12
lines changed

1 file changed

+17
-12
lines changed

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

+17-12
Original file line numberDiff line numberDiff line change
@@ -88,18 +88,23 @@ jobs:
8888
cuda-arch-list: '8.0'
8989
test-matrix: |
9090
{ 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" },
103108
]}
104109
selected-test-configs: ${{ inputs.benchmark_configs }}
105110
secrets: inherit

0 commit comments

Comments
 (0)