Skip to content

Commit e446372

Browse files
authored
Create self-scheduled-amd-mi355-caller.yml (#40134)
1 parent be1ab51 commit e446372

File tree

1 file changed

+63
-0
lines changed

1 file changed

+63
-0
lines changed
Lines changed: 63 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
name: Self-hosted runner scale set (AMD mi355 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 : amd-mi355-ci-1gpu
5+
# 2gpu : amd-mi355-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-mi355-ci
24+
docker: huggingface/transformers-pytorch-amd-gpu
25+
ci_event: Scheduled CI (AMD) - mi355
26+
report_repo_id: optimum-amd/transformers_daily_ci
27+
secrets: inherit
28+
29+
torch-pipeline:
30+
name: Torch pipeline CI
31+
uses: huggingface/hf-workflows/.github/workflows/transformers_amd_ci_scheduled_arc_scale_set.yaml@main
32+
with:
33+
job: run_pipelines_torch_gpu
34+
slack_report_channel: "#amd-hf-ci"
35+
runner_scale_set: amd-mi355-ci
36+
docker: huggingface/transformers-pytorch-amd-gpu
37+
ci_event: Scheduled CI (AMD) - mi355
38+
report_repo_id: optimum-amd/transformers_daily_ci
39+
secrets: inherit
40+
41+
example-ci:
42+
name: Example CI
43+
uses: huggingface/hf-workflows/.github/workflows/transformers_amd_ci_scheduled_arc_scale_set.yaml@main
44+
with:
45+
job: run_examples_gpu
46+
slack_report_channel: "#amd-hf-ci"
47+
runner_scale_set: amd-mi355-ci
48+
docker: huggingface/transformers-pytorch-amd-gpu
49+
ci_event: Scheduled CI (AMD) - mi355
50+
report_repo_id: optimum-amd/transformers_daily_ci
51+
secrets: inherit
52+
53+
deepspeed-ci:
54+
name: DeepSpeed CI
55+
uses: huggingface/hf-workflows/.github/workflows/transformers_amd_ci_scheduled_arc_scale_set.yaml@main
56+
with:
57+
job: run_torch_cuda_extensions_gpu
58+
slack_report_channel: "#amd-hf-ci"
59+
runner_scale_set: amd-mi355-ci
60+
docker: huggingface/transformers-pytorch-deepspeed-amd-gpu
61+
ci_event: Scheduled CI (AMD) - mi355
62+
report_repo_id: optimum-amd/transformers_daily_ci
63+
secrets: inherit

0 commit comments

Comments
 (0)