From 34e7e0e676332ccc4b3b222c9eca05787de50fc3 Mon Sep 17 00:00:00 2001 From: Akos Hadnagy Date: Thu, 4 Sep 2025 13:58:56 +0000 Subject: [PATCH 1/2] Change docker image to preview for the MI355 CI --- .github/workflows/self-scheduled-amd-mi355-caller.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/self-scheduled-amd-mi355-caller.yml b/.github/workflows/self-scheduled-amd-mi355-caller.yml index 19371aacf521..3eb06d78de7f 100644 --- a/.github/workflows/self-scheduled-amd-mi355-caller.yml +++ b/.github/workflows/self-scheduled-amd-mi355-caller.yml @@ -21,7 +21,7 @@ jobs: job: run_models_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: huggingface/transformers-pytorch-amd-gpu + docker: testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit @@ -33,7 +33,7 @@ jobs: job: run_pipelines_torch_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: huggingface/transformers-pytorch-amd-gpu + docker: testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit @@ -45,7 +45,7 @@ jobs: job: run_examples_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: huggingface/transformers-pytorch-amd-gpu + docker: testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit @@ -57,7 +57,7 @@ jobs: job: run_torch_cuda_extensions_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: huggingface/transformers-pytorch-deepspeed-amd-gpu + docker: testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit From b6fef57e3b1ce0106b2ed6f036489bf16fabaa00 Mon Sep 17 00:00:00 2001 From: Akos Hadnagy Date: Thu, 4 Sep 2025 14:59:15 +0000 Subject: [PATCH 2/2] Use pushed image --- .github/workflows/self-scheduled-amd-mi355-caller.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/self-scheduled-amd-mi355-caller.yml b/.github/workflows/self-scheduled-amd-mi355-caller.yml index 3eb06d78de7f..bd2fde3b0529 100644 --- a/.github/workflows/self-scheduled-amd-mi355-caller.yml +++ b/.github/workflows/self-scheduled-amd-mi355-caller.yml @@ -21,7 +21,7 @@ jobs: job: run_models_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: testing-rocm7.0-preview + docker: huggingface/testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit @@ -33,7 +33,7 @@ jobs: job: run_pipelines_torch_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: testing-rocm7.0-preview + docker: huggingface/testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit @@ -45,7 +45,7 @@ jobs: job: run_examples_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: testing-rocm7.0-preview + docker: huggingface/testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit @@ -57,7 +57,7 @@ jobs: job: run_torch_cuda_extensions_gpu slack_report_channel: "#amd-hf-ci" runner_scale_set: amd-mi355-ci - docker: testing-rocm7.0-preview + docker: huggingface/testing-rocm7.0-preview ci_event: Scheduled CI (AMD) - mi355 report_repo_id: optimum-amd/transformers_daily_ci secrets: inherit