Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
56 changes: 56 additions & 0 deletions .github/workflows/MI250_testing.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,56 @@
name: ROCM AMD mi250 scheduled CI caller

on:
workflow_dispatch:
workflow_run:
workflows: ["Self-hosted runner (AMD scheduled CI caller)"]
branches: ["main"]
types: [completed]
push:
branches:
- run_amd_scheduled_ci_caller*

jobs:
model-ci:
name: Model CI
uses: cemberk/hf-workflows/.github/workflows/transformers_amd_ci_scheduled.yaml@main

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should we establish a rocm/hf-workflows repository for this🤔? @Cemberk @gargrahul

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Agree, we should use ROCm fork instead of personal fork. Thanks.

with:
job: run_models_gpu
slack_report_channel: "#transformers-ci-daily-amd"
runner: mi250
docker: huggingface/transformers-pytorch-amd-gpu
ci_event: Scheduled CI (AMD) - mi250
secrets: inherit

torch-pipeline:
name: Torch pipeline CI
uses: cemberk/hf-workflows/.github/workflows/transformers_amd_ci_scheduled.yaml@main
with:
job: run_pipelines_torch_gpu
slack_report_channel: "#transformers-ci-daily-amd"

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Do we plan on adding an internal slack channel? If not, I think we can remove this line?

runner: mi250
docker: huggingface/transformers-pytorch-amd-gpu
ci_event: Scheduled CI (AMD) - mi250
secrets: inherit

example-ci:
name: Example CI
uses: cemberk/hf-workflows/.github/workflows/transformers_amd_ci_scheduled.yaml@main
with:
job: run_examples_gpu
slack_report_channel: "#transformers-ci-daily-amd"

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same here ^^

runner: mi250
docker: huggingface/transformers-pytorch-amd-gpu
ci_event: Scheduled CI (AMD) - mi250
secrets: inherit

deepspeed-ci:
name: DeepSpeed CI
uses: cemberk/hf-workflows/.github/workflows/transformers_amd_ci_scheduled.yaml@main
with:
job: run_torch_cuda_extensions_gpu
slack_report_channel: "#transformers-ci-daily-amd"
runner: mi250
docker: huggingface/transformers-pytorch-deepspeed-amd-gpu
ci_event: Scheduled CI (AMD) - mi250
secrets: inherit