File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -16,6 +16,10 @@ concurrency:
1616 cancel-in-progress : true
1717
1818jobs :
19+ docker-image :
20+ name : Resolve CI docker image
21+ uses : ./.github/workflows/_docker-image.yml
22+
1923 update-pytorch-commit-hash :
2024 runs-on : ubuntu-latest
2125 environment : ${{ (github.event_name == 'schedule') && 'update-commit-hash' || '' }}
5054 timeout : 180
5155
5256 test-static-hf-llm-qnn-linux :
57+ needs : docker-image
5358 name : test-static-hf-llm-qnn-linux
54- uses : pytorch/test-infra/.github/workflows/linux_job_v2 .yml@main
59+ uses : pytorch/test-infra/.github/workflows/linux_job_v3 .yml@main
5560 permissions :
5661 id-token : write
5762 contents : read
6065 task : [smollm2_135m]
6166 fail-fast : false
6267 with :
63- runner : linux.24xlarge
64- docker-image : ci-image:executorch-ubuntu-22.04-qnn-sdk
68+ runner : mt-l-x86iavx512-94-192
69+ docker-image : 308535385114.dkr.ecr.us-east-1.amazonaws.com/executorch/ ci-image:executorch-ubuntu-22.04-qnn-sdk-${{ needs.docker-image.outputs.ci-docker-hash }}
6570 submodules : ' recursive'
6671 ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
6772 timeout : 900
Original file line number Diff line number Diff line change @@ -21,6 +21,10 @@ concurrency:
2121permissions : read-all
2222
2323jobs :
24+ docker-image :
25+ name : Resolve CI docker image
26+ uses : ./.github/workflows/_docker-image.yml
27+
2428 gather-models :
2529 runs-on : ubuntu-22.04
2630 outputs :
@@ -42,17 +46,17 @@ jobs:
4246
4347 test-models-linux :
4448 name : test-models-linux
45- uses : pytorch/test-infra/.github/workflows/linux_job_v2 .yml@main
49+ uses : pytorch/test-infra/.github/workflows/linux_job_v3 .yml@main
4650 permissions :
4751 id-token : write
4852 contents : read
49- needs : gather-models
53+ needs : [docker-image, gather-models]
5054 strategy :
5155 matrix : ${{ fromJSON(needs.gather-models.outputs.models) }}
5256 fail-fast : false
5357 with :
5458 runner : ${{ matrix.runner }}
55- docker-image : ci-image:executorch-ubuntu-22.04-clang12
59+ docker-image : 308535385114.dkr.ecr.us-east-1.amazonaws.com/executorch/ ci-image:executorch-ubuntu-22.04-clang12-${{ needs.docker-image.outputs.ci-docker-hash }}
5660 submodules : ' recursive'
5761 ref : ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
5862 timeout : ${{ matrix.timeout }}
You can’t perform that action at this time.
0 commit comments