Skip to content

Commit 52dde84

Browse files
PaliCPaliC
committed
fix on
1 parent 37b9c8d commit 52dde84

File tree

1 file changed

+42
-185
lines changed

1 file changed

+42
-185
lines changed

‎.github/workflows/triton-inductor.yml

Lines changed: 42 additions & 185 deletions
Original file line numberDiff line numberDiff line change
@@ -16,199 +16,56 @@ on:
1616
pytorch_commit:
1717
description: 'Commit SHA to test PyTorch at'
1818
required: true
19-
default: 'viable/strict'
19+
default: 'main'
2020
schedule:
2121
# run nightly at 00:00 PST + random noise
2222
- cron: '15 7 * * *'
2323

2424
jobs:
25-
2625
linux-focal-rocm6_1-py3_8-inductor-build:
27-
name: rocm6.1-py3.8-inductor
28-
uses: pytorch/pytorch/.github/workflows/_linux-build.yml@main
29-
with:
30-
build-environment: linux-focal-rocm6.1-py3.8
31-
docker-image-name: pytorch-linux-focal-rocm-n-py3
32-
test-matrix: |
33-
{ include: [
34-
{ config: "inductor", shard: 1, num_shards: 2, runner: "linux.rocm.gpu.2" },
35-
{ config: "inductor", shard: 2, num_shards: 2, runner: "linux.rocm.gpu.2" },
36-
]}
37-
26+
steps:
27+
- name: Checkout pytorch-integration-testing
28+
uses: actions/checkout@v2
29+
with:
30+
path: pytorch-integration-testing
31+
32+
- name: Checkout PyTorch repository
33+
uses: actions/checkout@v2
34+
with:
35+
repository: pytorch/pytorch
36+
path: pytorch
37+
ref: {{inputs.pytorch_commit}}
38+
- name: rocm6.1-py3.8-inductor
39+
uses: pytorch/pytorch/.github/workflows/_linux-build.yml@main
40+
working-directory: pytorch
41+
with:
42+
build-environment: linux-focal-rocm6.1-py3.8
43+
docker-image-name: pytorch-linux-focal-rocm-n-py3
44+
test-matrix: |
45+
{ include: [
46+
{ config: "inductor", shard: 1, num_shards: 2, runner: "linux.rocm.gpu.2" },
47+
{ config: "inductor", shard: 2, num_shards: 2, runner: "linux.rocm.gpu.2" },
48+
]}
3849
linux-focal-rocm6_1-py3_8-inductor-test:
3950
permissions:
4051
id-token: write
4152
contents: read
42-
name: rocm6.1-py3.8-inductor
43-
uses: pytorch/pytorch/.github/workflows/_rocm-test.yml@main
4453
needs: linux-focal-rocm6_1-py3_8-inductor-build
45-
with:
46-
build-environment: linux-focal-rocm6.1-py3.8
47-
docker-image: ${{ needs.linux-focal-rocm6_1-py3_8-inductor-build.outputs.docker-image }}
48-
test-matrix: ${{ needs.linux-focal-rocm6_1-py3_8-inductor-build.outputs.test-matrix }}
49-
50-
linux-focal-cuda12_1-py3_10-gcc9-inductor-build:
51-
name: cuda12.1-py3.10-gcc9-sm86
52-
uses: pytorch/pytorch/.github/workflows/_linux-build.yml@main
53-
with:
54-
build-environment: linux-focal-cuda12.1-py3.10-gcc9-sm86
55-
docker-image-name: pytorch-linux-focal-cuda12.1-cudnn9-py3-gcc9-inductor-benchmarks
56-
cuda-arch-list: '8.6'
57-
test-matrix: |
58-
{ include: [
59-
{ config: "inductor", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
60-
{ config: "inductor", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
61-
{ config: "inductor_distributed", shard: 1, num_shards: 1, runner: "linux.g5.12xlarge.nvidia.gpu" },
62-
{ config: "inductor_huggingface", shard: 1, num_shards: 1, runner: "linux.g5.4xlarge.nvidia.gpu" },
63-
{ config: "inductor_timm", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
64-
{ config: "inductor_timm", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
65-
{ config: "inductor_torchbench", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
66-
{ config: "inductor_torchbench", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
67-
{ config: "dynamic_inductor_huggingface", shard: 1, num_shards: 1, runner: "linux.g5.4xlarge.nvidia.gpu" },
68-
{ config: "dynamic_inductor_timm", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
69-
{ config: "dynamic_inductor_timm", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
70-
{ config: "dynamic_inductor_torchbench", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
71-
{ config: "dynamic_inductor_torchbench", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
72-
{ config: "aot_inductor_huggingface", shard: 1, num_shards: 1, runner: "linux.g5.4xlarge.nvidia.gpu" },
73-
{ config: "aot_inductor_timm", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
74-
{ config: "aot_inductor_timm", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
75-
{ config: "aot_inductor_torchbench", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
76-
{ config: "aot_inductor_torchbench", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
77-
{ config: "inductor_cpp_wrapper_abi_compatible", shard: 1, num_shards: 1, runner: "linux.g5.4xlarge.nvidia.gpu" },
78-
]}
79-
secrets:
80-
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
81-
82-
linux-focal-cuda12_1-py3_10-gcc9-inductor-test:
83-
name: cuda12.1-py3.10-gcc9-sm86
84-
uses: pytorch/pytorch/.github/workflows/_linux-test.yml@main
85-
needs: linux-focal-cuda12_1-py3_10-gcc9-inductor-build
86-
with:
87-
build-environment: linux-focal-cuda12.1-py3.10-gcc9-sm86
88-
docker-image: ${{ needs.linux-focal-cuda12_1-py3_10-gcc9-inductor-build.outputs.docker-image }}
89-
test-matrix: ${{ needs.linux-focal-cuda12_1-py3_10-gcc9-inductor-build.outputs.test-matrix }}
90-
secrets:
91-
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
92-
93-
linux-focal-cuda12_1-py3_12-gcc9-inductor-build:
94-
name: cuda12.1-py3.12-gcc9-sm86
95-
uses: pytorch/pytorch/.github/workflows/_linux-build.yml@main
96-
with:
97-
build-environment: linux-focal-cuda12.1-py3.12-gcc9-sm86
98-
docker-image-name: pytorch-linux-focal-cuda12.1-cudnn9-py3.12-gcc9-inductor-benchmarks
99-
cuda-arch-list: '8.6'
100-
test-matrix: |
101-
{ include: [
102-
{ config: "inductor", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
103-
{ config: "inductor", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
104-
]}
105-
106-
linux-focal-cuda12_1-py3_12-gcc9-inductor-test:
107-
name: cuda12.1-py3.12-gcc9-sm86
108-
uses: pytorch/pytorch/.github/workflows/_linux-test.yml@main
109-
needs: linux-focal-cuda12_1-py3_12-gcc9-inductor-build
110-
with:
111-
build-environment: linux-focal-cuda12.1-py3.12-gcc9-sm86
112-
docker-image: ${{ needs.linux-focal-cuda12_1-py3_12-gcc9-inductor-build.outputs.docker-image }}
113-
test-matrix: ${{ needs.linux-focal-cuda12_1-py3_12-gcc9-inductor-build.outputs.test-matrix }}
114-
115-
linux-jammy-cpu-py3_12-inductor-halide-build:
116-
name: linux-jammy-cpu-py3.12-gcc11-inductor-halide
117-
uses: pytorch/pytorch/.github/workflows/_linux-build.yml@main
118-
with:
119-
build-environment: linux-jammy-py3.12-gcc11
120-
docker-image-name: pytorch-linux-jammy-py3.12-halide
121-
test-matrix: |
122-
{ include: [
123-
{ config: "inductor-halide", shard: 1, num_shards: 1, runner: "linux.12xlarge" },
124-
]}
125-
126-
linux-jammy-cpu-py3_12-inductor-halide-test:
127-
name: linux-jammy-cpu-py3.12-gcc11-inductor-halide
128-
uses: pytorch/pytorch/.github/workflows/_linux-test.yml@main
129-
needs: linux-jammy-cpu-py3_12-inductor-halide-build
130-
with:
131-
build-environment: linux-jammy-py3.12-gcc11
132-
docker-image: ${{ needs.linux-jammy-cpu-py3_12-inductor-halide-build.outputs.docker-image }}
133-
test-matrix: ${{ needs.linux-jammy-cpu-py3_12-inductor-halide-build.outputs.test-matrix }}
134-
135-
linux-focal-cuda12_4-py3_10-gcc9-inductor-build:
136-
# Should be synced with the one in inductor-periodic.yml but this only runs inductor_timm
137-
name: cuda12.4-py3.10-gcc9-sm86
138-
uses: pytorch/pytorch/.github/workflows/_linux-build.yml@main
139-
with:
140-
sync-tag: linux-focal-cuda12_4-py3_10-gcc9-inductor-build
141-
build-environment: linux-focal-cuda12.4-py3.10-gcc9-sm86
142-
docker-image-name: pytorch-linux-focal-cuda12.4-cudnn9-py3-gcc9-inductor-benchmarks
143-
cuda-arch-list: '8.6'
144-
test-matrix: |
145-
{ include: [
146-
{ config: "inductor_timm", shard: 1, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
147-
{ config: "inductor_timm", shard: 2, num_shards: 2, runner: "linux.g5.4xlarge.nvidia.gpu" },
148-
]}
149-
secrets:
150-
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
151-
152-
linux-focal-cuda12_4-py3_10-gcc9-inductor-test:
153-
name: cuda12.4-py3.10-gcc9-sm86
154-
uses: pytorch/pytorch/.github/workflows/_linux-test.yml@main
155-
needs: linux-focal-cuda12_4-py3_10-gcc9-inductor-build
156-
with:
157-
sync-tag: linux-focal-cuda12_4-py3_10-gcc9-inductor-test
158-
build-environment: linux-focal-cuda12.4-py3.10-gcc9-sm86
159-
docker-image: ${{ needs.linux-focal-cuda12_4-py3_10-gcc9-inductor-build.outputs.docker-image }}
160-
test-matrix: ${{ needs.linux-focal-cuda12_4-py3_10-gcc9-inductor-build.outputs.test-matrix }}
161-
secrets:
162-
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
163-
164-
linux-jammy-cpu-py3_8-gcc11-inductor-build:
165-
name: linux-jammy-cpu-py3.8-gcc11-inductor
166-
uses: pytorch/pytorch/.github/workflows/_linux-build.yml@main
167-
with:
168-
build-environment: linux-jammy-py3.8-gcc11-build
169-
docker-image-name: pytorch-linux-jammy-py3.8-gcc11-inductor-benchmarks
170-
test-matrix: |
171-
{ include: [
172-
{ config: "cpu_inductor_huggingface", shard: 1, num_shards: 1, runner: "linux.12xlarge" },
173-
{ config: "cpu_inductor_timm", shard: 1, num_shards: 2, runner: "linux.12xlarge" },
174-
{ config: "cpu_inductor_timm", shard: 2, num_shards: 2, runner: "linux.12xlarge" },
175-
{ config: "cpu_inductor_torchbench", shard: 1, num_shards: 2, runner: "linux.12xlarge" },
176-
{ config: "cpu_inductor_torchbench", shard: 2, num_shards: 2, runner: "linux.12xlarge" },
177-
{ config: "cpu_inductor_huggingface_freezing", shard: 1, num_shards: 1, runner: "linux.12xlarge" },
178-
{ config: "cpu_inductor_timm_freezing", shard: 1, num_shards: 2, runner: "linux.12xlarge" },
179-
{ config: "cpu_inductor_timm_freezing", shard: 2, num_shards: 2, runner: "linux.12xlarge" },
180-
{ config: "cpu_inductor_torchbench_freezing", shard: 1, num_shards: 2, runner: "linux.12xlarge" },
181-
{ config: "cpu_inductor_torchbench_freezing", shard: 2, num_shards: 2, runner: "linux.12xlarge" },
182-
{ config: "cpu_inductor_huggingface_amp_freezing", shard: 1, num_shards: 1, runner: "linux.16xlarge.spr" },
183-
{ config: "cpu_inductor_timm_amp_freezing", shard: 1, num_shards: 2, runner: "linux.16xlarge.spr" },
184-
{ config: "cpu_inductor_timm_amp_freezing", shard: 2, num_shards: 2, runner: "linux.16xlarge.spr" },
185-
{ config: "cpu_inductor_torchbench_amp_freezing", shard: 1, num_shards: 2, runner: "linux.16xlarge.spr" },
186-
{ config: "cpu_inductor_torchbench_amp_freezing", shard: 2, num_shards: 2, runner: "linux.16xlarge.spr" },
187-
{ config: "dynamic_cpu_inductor_huggingface", shard: 1, num_shards: 1, runner: "linux.12xlarge" },
188-
{ config: "dynamic_cpu_inductor_timm", shard: 1, num_shards: 2, runner: "linux.12xlarge" },
189-
{ config: "dynamic_cpu_inductor_timm", shard: 2, num_shards: 2, runner: "linux.12xlarge" },
190-
{ config: "dynamic_cpu_inductor_torchbench", shard: 1, num_shards: 2, runner: "linux.12xlarge" },
191-
{ config: "dynamic_cpu_inductor_torchbench", shard: 2, num_shards: 2, runner: "linux.12xlarge" },
192-
{ config: "cpu_aot_inductor_huggingface_freezing", shard: 1, num_shards: 1, runner: "linux.12xlarge" },
193-
{ config: "cpu_aot_inductor_timm_freezing", shard: 1, num_shards: 2, runner: "linux.12xlarge" },
194-
{ config: "cpu_aot_inductor_timm_freezing", shard: 2, num_shards: 2, runner: "linux.12xlarge" },
195-
{ config: "cpu_aot_inductor_torchbench_freezing", shard: 1, num_shards: 2, runner: "linux.12xlarge" },
196-
{ config: "cpu_aot_inductor_torchbench_freezing", shard: 2, num_shards: 2, runner: "linux.12xlarge" },
197-
{ config: "inductor_torchbench_cpu_smoketest_perf", shard: 1, num_shards: 1, runner: "linux.24xl.spr-metal" },
198-
]}
199-
secrets:
200-
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
201-
202-
linux-jammy-cpu-py3_8-gcc11-inductor-test:
203-
name: linux-jammy-cpu-py3.8-gcc11-inductor
204-
uses: pytorch/pytorch/.github/workflows/_linux-test.yml@main
205-
needs: linux-jammy-cpu-py3_8-gcc11-inductor-build
206-
with:
207-
build-environment: linux-jammy-py3.8-gcc11-build
208-
docker-image: ${{ needs.linux-jammy-cpu-py3_8-gcc11-inductor-build.outputs.docker-image }}
209-
test-matrix: ${{ needs.linux-jammy-cpu-py3_8-gcc11-inductor-build.outputs.test-matrix }}
210-
secrets:
211-
HUGGING_FACE_HUB_TOKEN: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}
212-
213-
214-
54+
steps:
55+
- name: Checkout pytorch-integration-testing
56+
uses: actions/checkout@v2
57+
with:
58+
path: pytorch-integration-testing
59+
60+
- name: Checkout PyTorch repository
61+
uses: actions/checkout@v2
62+
with:
63+
repository: pytorch/pytorch
64+
path: pytorch
65+
ref: {{inputs.pytorch_commit}}
66+
- name: rocm6.1-py3.8-inductor
67+
uses: pytorch/pytorch/.github/workflows/_rocm-test.yml@main
68+
with:
69+
build-environment: linux-focal-rocm6.1-py3.8
70+
docker-image: ${{ needs.linux-focal-rocm6_1-py3_8-inductor-build.outputs.docker-image }}
71+
test-matrix: ${{ needs.linux-focal-rocm6_1-py3_8-inductor-build.outputs.test-matrix }}

0 commit comments

Comments
 (0)