diff --git a/.github/workflows/model_jobs.yml b/.github/workflows/model_jobs.yml index 6ca019e7c93f..7e30cde735fa 100644 --- a/.github/workflows/model_jobs.yml +++ b/.github/workflows/model_jobs.yml @@ -25,6 +25,12 @@ on: required: false default: run_models_gpu type: string + runner_type: + required: false + type: string + report_repo_id: + required: false + type: string env: HF_HOME: /mnt/cache @@ -143,3 +149,15 @@ jobs: with: name: ${{ env.machine_type }}_${{ inputs.report_name_prefix }}_${{ env.matrix_folders }}_test_reports path: /transformers/reports/${{ env.machine_type }}_${{ inputs.report_name_prefix }}_${{ matrix.folders }}_test_reports + + collated_reports: + name: Collated Reports + if: ${{ always() }} + needs: run_models_gpu + uses: huggingface/transformers/.github/workflows/collated-reports.yml@main + with: + job: run_models_gpu + report_repo_id: ${{ inputs.report_repo_id }} + gpu_name: ${{ inputs.runner_type }} + machine_type: ${{ inputs.machine_type }} + secrets: inherit diff --git a/.github/workflows/self-scheduled-caller.yml b/.github/workflows/self-scheduled-caller.yml index d709c562251e..78c7f3c60f23 100644 --- a/.github/workflows/self-scheduled-caller.yml +++ b/.github/workflows/self-scheduled-caller.yml @@ -52,6 +52,7 @@ jobs: slack_report_channel: "#transformers-ci-daily-models" docker: huggingface/transformers-all-latest-gpu ci_event: Daily CI + runner_type: "a10" report_repo_id: hf-internal-testing/transformers_daily_ci commit_sha: ${{ github.sha }} secrets: inherit diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index 35d3e8d7b08e..a5dbc9d59a82 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -31,6 +31,9 @@ on: commit_sha: required: false type: string + runner_type: + required: false + type: string models: default: "" required: false @@ -126,6 +129,8 @@ jobs: runner_map: ${{ needs.setup.outputs.runner_map }} docker: ${{ inputs.docker }} commit_sha: ${{ inputs.commit_sha || github.sha }} + runner_type: ${{ inputs.runner_type }} + report_repo_id: ${{ inputs.report_repo_id }} secrets: inherit run_trainer_and_fsdp_gpu: