Add mi300 to amd daily ci workflows definition (#38415)

This commit is contained in:
ivarflakstad 2025-05-28 09:17:41 +02:00 committed by GitHub
parent 3b3ebcec40
commit 5f49e180a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1364,8 +1364,8 @@ if __name__ == "__main__":
nvidia_daily_ci_workflow = "huggingface/transformers/.github/workflows/self-scheduled-caller.yml"
amd_daily_ci_workflows = (
"huggingface/transformers/.github/workflows/self-scheduled-amd-mi210-caller.yml",
"huggingface/transformers/.github/workflows/self-scheduled-amd-mi250-caller.yml",
"huggingface/transformers/.github/workflows/self-scheduled-amd-mi300-caller.yml",
)
is_nvidia_daily_ci_workflow = os.environ.get("GITHUB_WORKFLOW_REF").startswith(nvidia_daily_ci_workflow)
is_amd_daily_ci_workflow = os.environ.get("GITHUB_WORKFLOW_REF").startswith(amd_daily_ci_workflows)