Skip to content

Commit 63f32a8

Browse files
Switch AMD daily CI to mi300 runners (#47259)
1 parent 1f8daee commit 63f32a8

5 files changed

Lines changed: 16 additions & 16 deletions

File tree

.github/workflows/benchmark_v2_mi325_caller.yml renamed to .github/workflows/benchmark_v2_mi300_caller.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
name: Benchmark v2 Scheduled Runner - MI325 Single-GPU
1+
name: Benchmark v2 Scheduled Runner - MI300 Single-GPU
22

33
on:
44
workflow_dispatch:
@@ -11,7 +11,7 @@ jobs:
1111
name: Benchmark v2 - Default Models
1212
uses: ./.github/workflows/benchmark_v2.yml
1313
with:
14-
runner: amd-mi325-ci-1gpu
14+
runner: amd-mi300-ci-1gpu
1515
container_image: huggingface/transformers-pytorch-amd-gpu
1616
container_options: --device /dev/kfd --device /dev/dri --env ROCR_VISIBLE_DEVICES --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache
1717
commit_sha: ${{ github.sha }}

.github/workflows/build-docker-images.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ jobs:
217217
name: "Cache Latest Pytorch (AMD) Image"
218218
needs: latest-pytorch-amd
219219
runs-on:
220-
group: amd-mi325-1gpu
220+
group: hfc-amd-mi300-1gpu
221221
steps:
222222
-
223223
name: Login to DockerHub

.github/workflows/self-scheduled-amd-mi325-caller.yml renamed to .github/workflows/self-scheduled-amd-mi300-caller.yml

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
1-
name: Self-hosted runner scale set (AMD mi325 scheduled CI caller)
1+
name: Self-hosted runner scale set (AMD mi300 scheduled CI caller)
22

33
# 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-mi325-ci-1gpu
5-
# 2gpu scale set: amd-mi325-ci-2gpu
4+
# For example, 1gpu scale set: amd-mi300-ci-1gpu
5+
# 2gpu scale set: amd-mi300-ci-2gpu
66
# Important: Do not pin the reusable workflow ref to a SHA. AMD runner groups only route jobs for
77
# workflows referenced at @main; a pinned SHA causes jobs to wait for a runner until the 24h timeout.
88

@@ -25,9 +25,9 @@ jobs:
2525
with:
2626
job: run_models_gpu
2727
slack_report_channel: "#amd-hf-ci"
28-
runner_group: amd-mi325
28+
runner_group: hfc-amd-mi300
2929
docker: huggingface/transformers-pytorch-amd-gpu
30-
ci_event: Scheduled CI (AMD) - mi325
30+
ci_event: Scheduled CI (AMD) - mi300
3131
report_repo_id: optimum-amd/transformers_daily_ci
3232
env_file: /etc/podinfo/gha-gpu-isolation-settings
3333
secrets: inherit
@@ -38,9 +38,9 @@ jobs:
3838
with:
3939
job: run_pipelines_torch_gpu
4040
slack_report_channel: "#amd-hf-ci"
41-
runner_group: amd-mi325
41+
runner_group: hfc-amd-mi300
4242
docker: huggingface/transformers-pytorch-amd-gpu
43-
ci_event: Scheduled CI (AMD) - mi325
43+
ci_event: Scheduled CI (AMD) - mi300
4444
report_repo_id: optimum-amd/transformers_daily_ci
4545
env_file: /etc/podinfo/gha-gpu-isolation-settings
4646
secrets: inherit
@@ -51,9 +51,9 @@ jobs:
5151
with:
5252
job: run_examples_gpu
5353
slack_report_channel: "#amd-hf-ci"
54-
runner_group: amd-mi325
54+
runner_group: hfc-amd-mi300
5555
docker: huggingface/transformers-pytorch-amd-gpu
56-
ci_event: Scheduled CI (AMD) - mi325
56+
ci_event: Scheduled CI (AMD) - mi300
5757
report_repo_id: optimum-amd/transformers_daily_ci
5858
env_file: /etc/podinfo/gha-gpu-isolation-settings
5959
secrets: inherit
@@ -64,9 +64,9 @@ jobs:
6464
with:
6565
job: run_torch_cuda_extensions_gpu
6666
slack_report_channel: "#amd-hf-ci"
67-
runner_group: amd-mi325
67+
runner_group: hfc-amd-mi300
6868
docker: huggingface/transformers-pytorch-deepspeed-amd-gpu
69-
ci_event: Scheduled CI (AMD) - mi325
69+
ci_event: Scheduled CI (AMD) - mi300
7070
report_repo_id: optimum-amd/transformers_daily_ci
7171
env_file: /etc/podinfo/gha-gpu-isolation-settings
7272
secrets: inherit

utils/collated_reports.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from pathlib import Path
2020

2121

22-
DEFAULT_GPU_NAMES = ["mi300", "mi325", "mi355", "h100", "a10"]
22+
DEFAULT_GPU_NAMES = ["mi300", "mi355", "h100", "a10"]
2323

2424

2525
def simplify_gpu_name(gpu_name: str, simplified_names: list[str]) -> str:

utils/notification_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1411,7 +1411,7 @@ def pop_default(l: list[Any], i: int, default: Any) -> Any:
14111411
"huggingface/transformers/.github/workflows/self-scheduled-flash-attn-caller.yml",
14121412
)
14131413
amd_daily_ci_workflows = (
1414-
"huggingface/transformers/.github/workflows/self-scheduled-amd-mi325-caller.yml",
1414+
"huggingface/transformers/.github/workflows/self-scheduled-amd-mi300-caller.yml",
14151415
"huggingface/transformers/.github/workflows/self-scheduled-amd-mi355-caller.yml",
14161416
)
14171417
is_nvidia_daily_ci_workflow = os.environ.get("GITHUB_WORKFLOW_REF").startswith(nvidia_daily_ci_workflow)

0 commit comments

Comments
 (0)