diff --git a/.github/workflows/pull.yml b/.github/workflows/pull.yml index c4e6c7fb9e7..b672d691446 100644 --- a/.github/workflows/pull.yml +++ b/.github/workflows/pull.yml @@ -895,8 +895,9 @@ jobs: backends/arm/test/test_arm_backend.sh "${ARM_TEST}" test-arm-backend-vkml: + needs: docker-image name: test-arm-backend-vkml - 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 @@ -908,8 +909,8 @@ jobs: - test_arm_backend: test_runtime_vgf fail-fast: false with: - runner: linux.2xlarge.memory - docker-image: ci-image:executorch-ubuntu-24.04-arm-sdk + runner: mt-l-x86iavx512-8-64 + docker-image: ${{ needs.docker-image.outputs.docker-registry }}/ci-image:executorch-ubuntu-24.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: 120 @@ -925,8 +926,6 @@ jobs: --enable-mlsdk-deps \ --install-mlsdk-deps-with-pip - sudo sysctl fs.inotify.max_user_watches=1048576 - ARM_TEST=${{ matrix.test_arm_backend }} if [[ -n "${RUNNER_TEST_RESULTS_DIR:-}" ]]; then