Skip to content

Commit c4e71e8

Browse files
authored
Add AMD MI300 CI caller leveraging self-hosted runner scale set workflow in hf-workflows (#38132)
1 parent 706b009 commit c4e71e8

File tree

1 file changed

+59
-0
lines changed

1 file changed

+59
-0
lines changed
Lines changed: 59 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,59 @@
1+
name: Self-hosted runner scale set (AMD mi300 scheduled CI caller)
2+
3+
# Note: For every job in this workflow, the name of the runner scale set is finalized in the runner yaml i.e. huggingface/hf-workflows/.github/workflows/transformers_amd_ci_scheduled_arc_scale_set.yaml
4+
# For example, 1gpu scale set: amd-mi300-ci-1gpu
5+
# 2gpu scale set: amd-mi300-ci-2gpu
6+
7+
on:
8+
workflow_run:
9+
workflows: ["Self-hosted runner (AMD scheduled CI caller)"]
10+
branches: ["main"]
11+
types: [completed]
12+
push:
13+
branches:
14+
- run_amd_scheduled_ci_caller*
15+
16+
jobs:
17+
model-ci:
18+
name: Model CI
19+
uses: huggingface/hf-workflows/.github/workflows/transformers_amd_ci_scheduled_arc_scale_set.yaml@main
20+
with:
21+
job: run_models_gpu
22+
slack_report_channel: "#amd-hf-ci"
23+
runner_scale_set: amd-mi300-ci
24+
docker: huggingface/transformers-pytorch-amd-gpu
25+
ci_event: Scheduled CI (AMD) - mi300
26+
secrets: inherit
27+
28+
torch-pipeline:
29+
name: Torch pipeline CI
30+
uses: huggingface/hf-workflows/.github/workflows/transformers_amd_ci_scheduled_arc_scale_set.yaml@main
31+
with:
32+
job: run_pipelines_torch_gpu
33+
slack_report_channel: "#amd-hf-ci"
34+
runner_scale_set: amd-mi300-ci
35+
docker: huggingface/transformers-pytorch-amd-gpu
36+
ci_event: Scheduled CI (AMD) - mi300
37+
secrets: inherit
38+
39+
example-ci:
40+
name: Example CI
41+
uses: huggingface/hf-workflows/.github/workflows/transformers_amd_ci_scheduled_arc_scale_set.yaml@main
42+
with:
43+
job: run_examples_gpu
44+
slack_report_channel: "#amd-hf-ci"
45+
runner_scale_set: amd-mi300-ci
46+
docker: huggingface/transformers-pytorch-amd-gpu
47+
ci_event: Scheduled CI (AMD) - mi300
48+
secrets: inherit
49+
50+
deepspeed-ci:
51+
name: DeepSpeed CI
52+
uses: huggingface/hf-workflows/.github/workflows/transformers_amd_ci_scheduled_arc_scale_set.yaml@main
53+
with:
54+
job: run_torch_cuda_extensions_gpu
55+
slack_report_channel: "#amd-hf-ci"
56+
runner_scale_set: amd-mi300-ci
57+
docker: huggingface/transformers-pytorch-deepspeed-amd-gpu
58+
ci_event: Scheduled CI (AMD) - mi300
59+
secrets: inherit

0 commit comments

Comments
 (0)