Skip to content
Draft
Show file tree
Hide file tree
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
11 changes: 8 additions & 3 deletions .github/workflows/_llm_server.yml
Original file line number Diff line number Diff line change
Expand Up @@ -10,14 +10,19 @@ on:
default: ci-image:executorch-ubuntu-22.04-clang12

jobs:
docker-image:
name: Resolve CI docker image
uses: ./.github/workflows/_docker-image.yml

linux:
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
needs: docker-image
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
permissions:
id-token: write
contents: read
with:
runner: linux.2xlarge
docker-image: ${{ inputs.docker-image }}
runner: mt-l-x86iavx512-8-64
docker-image: ${{ needs.docker-image.outputs.docker-registry }}/${{ inputs.docker-image }}-${{ needs.docker-image.outputs.ci-docker-hash }}
submodules: recursive
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
timeout: 60
Expand Down
11 changes: 8 additions & 3 deletions .github/workflows/_test_arduino_library.yml
Original file line number Diff line number Diff line change
Expand Up @@ -14,15 +14,20 @@ on:
default: 90

jobs:
docker-image:
name: Resolve CI docker image
uses: ./.github/workflows/_docker-image.yml

run:
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
needs: docker-image
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
permissions:
id-token: write
contents: read
with:
job-name: arduino-library
runner: linux.2xlarge
docker-image: ci-image:executorch-ubuntu-22.04-arm-sdk
runner: mt-l-x86iavx512-8-64
docker-image: ${{ needs.docker-image.outputs.docker-registry }}/ci-image:executorch-ubuntu-22.04-arm-sdk-${{ needs.docker-image.outputs.ci-docker-hash }}
submodules: 'recursive'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
timeout: ${{ inputs.timeout }}
Expand Down
14 changes: 11 additions & 3 deletions .github/workflows/_test_backend.yml
Original file line number Diff line number Diff line change
Expand Up @@ -40,15 +40,20 @@ on:
description: 'Runner type for Linux jobs'
required: false
type: string
default: linux.4xlarge.memory
default: mt-l-x86iavx512-16-128
docker-image:
description: 'Docker image for Linux jobs'
required: false
type: string
default: ci-image:executorch-ubuntu-22.04-clang12

jobs:
docker-image:
name: Resolve CI docker image
uses: ./.github/workflows/_docker-image.yml

test-backend-linux:
needs: docker-image
if: ${{ inputs.run-linux }}
strategy:
fail-fast: false
Expand All @@ -57,11 +62,14 @@ jobs:
suite: [models, operators]
exclude: ${{ fromJSON(inputs.exclude) }}

uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
permissions:
id-token: write
contents: read
with:
ref: ${{ inputs.ref }}
runner: ${{ inputs.runner-linux }}
docker-image: ${{ inputs.docker-image }}
docker-image: ${{ needs.docker-image.outputs.docker-registry }}/${{ inputs.docker-image }}-${{ needs.docker-image.outputs.ci-docker-hash }}
submodules: recursive
timeout: ${{ inputs.timeout }}
upload-artifact: test-report-${{ inputs.backend }}-${{ matrix.flow }}-${{ matrix.suite }}
Expand Down
22 changes: 17 additions & 5 deletions .github/workflows/_test_cadence.yml
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ on:
description: 'Runner type'
required: false
type: string
default: linux.8xlarge.memory
default: mt-l-x86iavx512-32-256
ref:
description: 'Git ref to checkout'
required: false
Expand All @@ -29,12 +29,20 @@ on:
default: 90

jobs:
docker-image:
name: Resolve CI docker image
uses: ./.github/workflows/_docker-image.yml

test-aot:
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
needs: docker-image
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
permissions:
id-token: write
contents: read
with:
job-name: test-aot
runner: ${{ inputs.runner }}
docker-image: ${{ inputs.docker-image }}
docker-image: ${{ needs.docker-image.outputs.docker-registry }}/${{ inputs.docker-image }}-${{ needs.docker-image.outputs.ci-docker-hash }}
submodules: recursive
ref: ${{ inputs.ref }}
timeout: ${{ inputs.timeout }}
Expand All @@ -50,11 +58,15 @@ jobs:
python -m pytest backends/cadence/aot/tests/ -v -n auto --reruns 2 --reruns-delay 1

test-ops:
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
needs: docker-image
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
permissions:
id-token: write
contents: read
with:
job-name: test-ops
runner: ${{ inputs.runner }}
docker-image: ${{ inputs.docker-image }}
docker-image: ${{ needs.docker-image.outputs.docker-registry }}/${{ inputs.docker-image }}-${{ needs.docker-image.outputs.ci-docker-hash }}
submodules: recursive
ref: ${{ inputs.ref }}
timeout: ${{ inputs.timeout }}
Expand Down
14 changes: 11 additions & 3 deletions .github/workflows/_test_cortex_m_e2e.yml
Original file line number Diff line number Diff line change
Expand Up @@ -23,17 +23,25 @@ on:
default: 120

jobs:
docker-image:
name: Resolve CI docker image
uses: ./.github/workflows/_docker-image.yml

run:
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
needs: docker-image
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
permissions:
id-token: write
contents: read
strategy:
matrix:
model: ${{ fromJSON(inputs.models) }}
target: ${{ fromJSON(inputs.targets) }}
fail-fast: false
with:
job-name: ${{ matrix.model }}-${{ matrix.target }}
runner: linux.2xlarge.memory
docker-image: ci-image:executorch-ubuntu-22.04-arm-sdk
runner: mt-l-x86iavx512-8-64
docker-image: ${{ needs.docker-image.outputs.docker-registry }}/ci-image:executorch-ubuntu-22.04-arm-sdk-${{ needs.docker-image.outputs.ci-docker-hash }}
submodules: 'recursive'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
timeout: ${{ inputs.timeout }}
Expand Down
14 changes: 11 additions & 3 deletions .github/workflows/_test_cortex_m_ops.yml
Original file line number Diff line number Diff line change
Expand Up @@ -18,16 +18,24 @@ on:
default: 120

jobs:
docker-image:
name: Resolve CI docker image
uses: ./.github/workflows/_docker-image.yml

run:
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
needs: docker-image
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
permissions:
id-token: write
contents: read
strategy:
matrix:
target: ${{ fromJSON(inputs.targets) }}
fail-fast: false
with:
job-name: cortex-m-ops-${{ matrix.target }}
runner: linux.2xlarge.memory
docker-image: ci-image:executorch-ubuntu-22.04-arm-sdk
runner: mt-l-x86iavx512-8-64
docker-image: ${{ needs.docker-image.outputs.docker-registry }}/ci-image:executorch-ubuntu-22.04-arm-sdk-${{ needs.docker-image.outputs.ci-docker-hash }}
submodules: 'recursive'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
timeout: ${{ inputs.timeout }}
Expand Down
14 changes: 11 additions & 3 deletions .github/workflows/_test_riscv.yml
Original file line number Diff line number Diff line change
Expand Up @@ -37,11 +37,19 @@ on:
type: string

jobs:
docker-image:
name: Resolve CI docker image
uses: ./.github/workflows/_docker-image.yml

run:
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
needs: docker-image
uses: pytorch/test-infra/.github/workflows/linux_job_v3.yml@main
permissions:
id-token: write
contents: read
with:
runner: linux.2xlarge
docker-image: ci-image:executorch-ubuntu-24.04-gcc14
runner: mt-l-x86iavx512-8-64
docker-image: ${{ needs.docker-image.outputs.docker-registry }}/ci-image:executorch-ubuntu-24.04-gcc14-${{ needs.docker-image.outputs.ci-docker-hash }}
submodules: 'recursive'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
timeout: ${{ inputs.timeout }}
Expand Down
6 changes: 6 additions & 0 deletions .github/workflows/test-backend-arm.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,9 @@ concurrency:
jobs:
test-arm:
uses: ./.github/workflows/_test_backend.yml
permissions:
id-token: write
contents: read
with:
backend: arm
flows: >-
Expand All @@ -35,6 +38,9 @@ jobs:

test-arm-vgf:
uses: ./.github/workflows/_test_backend.yml
permissions:
id-token: write
contents: read
with:
backend: arm-vgf
flows: >-
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/test-backend-coreml.yml
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,9 @@ jobs:
contains(needs.changed-files.outputs.changed-files, '.github/workflows/test-backend-coreml.yml') ||
contains(needs.changed-files.outputs.changed-files, '.github/workflows/_test_backend.yml')
uses: ./.github/workflows/_test_backend.yml
permissions:
id-token: write
contents: read
with:
backend: coreml
# The heavier coreml_static_int8 macOS matrix saturates the shared
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/test-backend-cortex-m.yml
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,9 @@ jobs:
contains(needs.changed-files.outputs.changed-files, '.github/workflows/test-backend-cortex-m.yml') ||
contains(needs.changed-files.outputs.changed-files, '.github/workflows/_test_backend.yml')
uses: ./.github/workflows/_test_backend.yml
permissions:
id-token: write
contents: read
with:
backend: cortex_m
flows: '["cortex_m"]'
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/test-backend-nxp.yml
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,9 @@ jobs:
contains(needs.changed-files.outputs.changed-files, '.github/workflows/test-backend-nxp.yml') ||
contains(needs.changed-files.outputs.changed-files, '.github/workflows/_test_backend.yml')
uses: ./.github/workflows/_test_backend.yml
permissions:
id-token: write
contents: read
with:
backend: nxp
flows: '["nxp_neutron_imxrt700_int8_ptq"]'
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/test-backend-openvino.yml
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,9 @@ concurrency:
jobs:
test-openvino:
uses: ./.github/workflows/_test_backend.yml
permissions:
id-token: write
contents: read
with:
backend: openvino
flows: '["openvino"]'
Expand Down
5 changes: 4 additions & 1 deletion .github/workflows/test-backend-qnn.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,9 @@ concurrency:
jobs:
test-qnn:
uses: ./.github/workflows/_test_backend.yml
permissions:
id-token: write
contents: read
with:
backend: qnn
flows: >-
Expand All @@ -28,4 +31,4 @@ jobs:
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
timeout: 120
run-linux: true
runner-linux: linux.8xlarge.memory
runner-linux: mt-l-x86iavx512-32-256
3 changes: 3 additions & 0 deletions .github/workflows/test-backend-vulkan.yml
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,9 @@ jobs:
# runners. Runs on every PR and nightly.
test-vulkan:
uses: ./.github/workflows/_test_backend.yml
permissions:
id-token: write
contents: read
with:
backend: vulkan
flows: >-
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/test-backend-webgpu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,9 @@ concurrency:
jobs:
test-webgpu:
uses: ./.github/workflows/_test_backend.yml
permissions:
id-token: write
contents: read
with:
backend: webgpu
flows: '["webgpu"]'
Expand Down
3 changes: 3 additions & 0 deletions .github/workflows/test-backend-xnnpack.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,9 @@ concurrency:
jobs:
test-xnnpack:
uses: ./.github/workflows/_test_backend.yml
permissions:
id-token: write
contents: read
with:
backend: xnnpack
flows: >-
Expand Down
Loading