diff --git a/.github/workflows/_test_backend.yml b/.github/workflows/_test_backend.yml index 6323c007f4a..11a34bfa4fb 100644 --- a/.github/workflows/_test_backend.yml +++ b/.github/workflows/_test_backend.yml @@ -11,6 +11,11 @@ on: description: 'JSON array of flows to test' required: true type: string + matrix-exclude: + description: 'JSON array of {flow, suite} combos to exclude from the matrix' + required: false + type: string + default: '[]' ref: description: 'Git ref to checkout' required: false @@ -50,6 +55,7 @@ jobs: matrix: flow: ${{ fromJSON(inputs.flows) }} suite: [models, operators] + exclude: ${{ fromJSON(inputs.matrix-exclude) }} uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main with: @@ -126,6 +132,7 @@ jobs: matrix: flow: ${{ fromJSON(inputs.flows) }} suite: [models, operators] + exclude: ${{ fromJSON(inputs.matrix-exclude) }} uses: pytorch/test-infra/.github/workflows/macos_job.yml@main with: diff --git a/.github/workflows/test-backend-coreml.yml b/.github/workflows/test-backend-coreml.yml index 19f77acad29..31396fd5140 100644 --- a/.github/workflows/test-backend-coreml.yml +++ b/.github/workflows/test-backend-coreml.yml @@ -42,6 +42,9 @@ jobs: ${{ github.event_name == 'pull_request' && '["coreml"]' || '["coreml", "coreml_static_int8"]' }} + # Run coreml_static_int8 for the operators suite only; skip it for + # the models suite (still covered by the coreml flow). + matrix-exclude: '[{"flow": "coreml_static_int8", "suite": "models"}]' ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} timeout: 180 run-macos: true