8000 Integrated AMD AWS runners into Pytorch CI · pytorch/pytorch@12d181b · GitHub
[go: up one dir, main page]
Skip to content

Commit 12d181b

Browse files
charan-ponnadacharan-ponnada
committed
Integrated AMD AWS runners into Pytorch CI
Integrated AMD AWS runners into PyTorch CI, including the linux.24xl.amd for performance tests, the linux.12xl.amd with AVX512 support for unit and periodic tests, and the linux.8xl.amd with AVX2 support for unit and periodic tests. Signed-off-by: charan-ponnada <charan.ponnada@amd.com> Signed-off-by: kiriti-pendyala <kiriti.pendyala@amd.com>
1 parent 3443627 commit 12d181b

File tree

4 files changed

+32
-0
lines changed

4 files changed

+32
-0
lines changed

‎.github/workflows/inductor-micro-benchmark-x86.yml

+1
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ jobs:
2727
test-matrix: |
2828
{ include: [
2929
{ config: "inductor-micro-benchmark-cpu-x86", shard: 1, num_shards: 1, runner: "linux.24xl.spr-metal", owners: ["oncall:pt2"] },
30+
{ config: "inductor-micro-benchmark-cpu-x86", shard: 1, num_shards: 1, runner: "linux.24xlarge.amd", owners: ["oncall:pt2"] },
3031
]}
3132
secrets: inherit
3233

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

+12
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,18 @@ jobs:
8686
{ config: "inductor_torchbench_perf_cpu_x86", shard: 2, num_shards: 4, runner: "linux.24xl.spr-metal" },
8787
{ config: "inductor_torchbench_perf_cpu_x86", shard: 3, num_shards: 4, runner: "linux.24xl.spr-metal" },
8888
{ config: "inductor_torchbench_perf_cpu_x86", shard: 4, num_shards: 4, runner: "linux.24xl.spr-metal" },
89+
{ config: "inductor_huggingface_perf_cpu_x86", shard: 1, num_shards: 3, runner: "linux.24xlarge.amd" },
90+
{ config: "inductor_huggingface_perf_cpu_x86", shard: 2, num_shards: 3, runner: "linux.24xlarge.amd" },
91+
{ config: "inductor_huggingface_perf_cpu_x86", shard: 3, num_shards: 3, runner: "linux.24xlarge.amd" },
92+
{ config: "inductor_timm_perf_cpu_x86", shard: 1, num_shards: 5, runner: "linux.24xlarge.amd" },
93+
{ config: "inductor_timm_perf_cpu_x86", shard: 2, num_shards: 5, runner: "linux.24xlarge.amd" },
94+
{ config: "inductor_timm_perf_cpu_x86", shard: 3, num_shards: 5, runner: "linux.24xlarge.amd" },
95+
{ config: "inductor_timm_perf_cpu_x86", shard: 4, num_shards: 5, runner: "linux.24xlarge.amd" },
96+
{ config: "inductor_timm_perf_cpu_x86", shard: 5, num_shards: 5, runner: "linux.24xlarge.amd" },
97+
{ config: "inductor_torchbench_perf_cpu_x86", shard: 1, num_shards: 4, runner: "linux.24xlarge.amd" },
98+
{ config: "inductor_torchbench_perf_cpu_x86", shard: 2, num_shards: 4, runner: "linux.24xlarge.amd" },
99+
{ config: "inductor_torchbench_perf_cpu_x86", shard: 3, num_shards: 4, runner: "linux.24xlarge.amd" },
100+
{ config: "inductor_torchbench_perf_cpu_x86", shard: 4, num_shards: 4, runner: "linux.24xlarge.amd" },
89101
]}
90102
selected-test-configs: ${{ inputs.benchmark_configs }}
91103
secrets: inherit

‎.github/workflows/inductor-periodic.yml

+11
Original file line numberDiff line numberDiff line change
@@ -156,6 +156,17 @@ jobs:
156156
{ config: "cpu_inductor_freezing_avx2_torchbench", shard: 2, num_shards: 2, runner: "linux.10xlarge.avx2" },
157157
{ config: "cpu_inductor_freezing_avx2_timm", shard: 1, num_shards: 2, runner: "linux.10xlarge.avx2" },
158158
{ config: "cpu_inductor_freezing_avx2_timm", shard: 2, num_shards: 2, runner: "linux.10xlarge.avx2" },
159+
{ config: "cpu_inductor_huggingface", shard: 1, num_shards: 1, runner: "linux.8xlarge.amd" },
160+
{ config: "cpu_inductor_timm", shard: 1, num_shards: 2, runner: "linux.8xlarge.amd" },
161+
{ config: "cpu_inductor_timm", shard: 2, num_shards: 2, runner: "linux.8xlarge.amd" },
162+
{ config: "dynamic_cpu_inductor_huggingface", shard: 1, num_shards: 1, runner: "linux.8xlarge.amd" },
163+
{ config: "dynamic_cpu_inductor_timm", shard: 1, num_shards: 2, runner: "linux.8xlarge.amd" },
164+
{ config: "dynamic_cpu_inductor_timm", shard: 2, num_shards: 2, runner: "linux.8xlarge.amd" },
165+
{ config: "cpu_inductor_freezing_avx2_huggingface", shard: 1, num_shards: 1, runner: "linux.12xlarge.amd" },
166+
{ config: "cpu_inductor_freezing_avx2_torchbench", shard: 1, num_shards: 2, runner: "linux.12xlarge.amd" },
167+
{ config: "cpu_inductor_freezing_avx2_torchbench", shard: 2, num_shards: 2, runner: "linux.12xlarge.amd" },
168+
{ config: "cpu_inductor_freezing_avx2_timm", shard: 1, num_shards: 2, runner: "linux.12xlarge.amd" },
169+
{ config: "cpu_inductor_freezing_avx2_timm", shard: 2, num_shards: 2, runner: "linux.12xlarge.amd" },
159170
]}
160171
secrets: inherit
161172

‎.github/workflows/inductor.yml

+8
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,14 @@ jobs:
9191
{ config: "dynamic_cpu_inductor_torchbench", shard: 1, num_shards: 2, runner: "${{ needs.get-label-type.outputs.label-type }}linux.8xlarge.amx" },
9292
{ config: "dynamic_cpu_inductor_torchbench", shard: 2, num_shards: 2, runner: "${{ needs.get-label-type.outputs.label-type }}linux.8xlarge.amx" },
9393
{ config: "inductor_torchbench_cpu_smoketest_perf", shard: 1, num_shards: 1, runner: "${{ needs.get-label-type.outputs.label-type }}linux.24xl.spr-metal" },
94+
{ config: "cpu_inductor_torchbench", shard: 1, num_shards: 2, runner: "linux.8xlarge.amd" },
95+
{ config: "cpu_inductor_torchbench", shard: 2, num_shards: 2, runner: "linux.8xlarge.amd" },
96+
{ config: "dynamic_cpu_inductor_huggingface", shard: 1, num_shards: 1, runner: "linux.8xlarge.amd" },
97+
{ config: "dynamic_cpu_inductor_timm", shard: 1, num_shards: 2, runner: "linux.8xlarge.amd" },
98+
{ config: "dynamic_cpu_inductor_timm", shard: 2, num_shards: 2, runner: "linux.8xlarge.amd" },
99+
{ config: "dynamic_cpu_inductor_torchbench", shard: 1, num_shards: 2, runner: "linux.8xlarge.amd" },
100+
{ config: "dynamic_cpu_inductor_torchbench", shard: 2, num_shards: 2, runner: "linux.8xlarge.amd" },
101+
{ config: "inductor_torchbench_cpu_smoketest_perf", shard: 1, num_shards: 1, runner: "linux.24xl.amd" },
94102
]}
95103
secrets: inherit
96104

0 commit comments

Comments
 (0)
0