Skip to content

Commit

Permalink
Replace runners prefix amz2023.
Browse files Browse the repository at this point in the history
  • Loading branch information
jeanschmidt committed Jul 25, 2024
1 parent dbf7d6e commit 7d1746b
Show file tree
Hide file tree
Showing 15 changed files with 272 additions and 272 deletions.
18 changes: 9 additions & 9 deletions .ci/scripts/gather_test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,19 +14,19 @@
from examples.xnnpack import MODEL_NAME_TO_OPTIONS

DEFAULT_RUNNERS = {
"linux": "linux.2xlarge",
"linux": "amz2023.linux.2xlarge",
"macos": "macos-m1-stable",
}
CUSTOM_RUNNERS = {
"linux": {
# This one runs OOM on smaller runner, the root cause is unclear (T163016365)
"w2l": "linux.12xlarge",
"ic4": "linux.12xlarge",
"resnet50": "linux.12xlarge",
"llava": "linux.12xlarge",
"w2l": "amz2023.linux.12xlarge",
"ic4": "amz2023.linux.12xlarge",
"resnet50": "amz2023.linux.12xlarge",
"llava": "amz2023.linux.12xlarge",
# This one causes timeout on smaller runner, the root cause is unclear (T161064121)
"dl3": "linux.12xlarge",
"emformer_join": "linux.12xlarge",
"dl3": "amz2023.linux.12xlarge",
"emformer_join": "amz2023.linux.12xlarge",
}
}

Expand Down Expand Up @@ -116,7 +116,7 @@ def export_models_for_ci() -> dict[str, dict]:
"build-tool": "buck2",
"model": "mv3",
"backend": backend,
"runner": "linux.2xlarge",
"runner": "amz2023.linux.2xlarge",
"timeout": DEFAULT_TIMEOUT,
}
models["include"].append(record)
Expand Down Expand Up @@ -145,7 +145,7 @@ def export_models_for_ci() -> dict[str, dict]:
"build-tool": "cmake",
"model": name,
"backend": backend,
"runner": DEFAULT_RUNNERS.get(target_os, "linux.2xlarge"),
"runner": DEFAULT_RUNNERS.get(target_os, "amz2023.linux.2xlarge"),
"timeout": DEFAULT_TIMEOUT,
}

Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/_unittest.yml
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ jobs:
linux:
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: ${{ inputs.docker-image }}
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand Down
8 changes: 4 additions & 4 deletions .github/workflows/android.yml
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ jobs:
matrix:
tokenizer: [bpe, tiktoken]
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-clang12-android
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand All @@ -49,7 +49,7 @@ jobs:
# Upload artifacts to S3. The artifacts are needed not only by the device farm but also TorchChat
upload-artifacts:
needs: build-llm-demo
runs-on: linux.2xlarge
runs-on: amz2023.linux.2xlarge
steps:
- name: Download the artifacts from GitHub
uses: actions/download-artifact@v3
Expand Down Expand Up @@ -81,7 +81,7 @@ jobs:
# Running Android emulator directly on the runner and not using Docker
run-emulator:
needs: build-llm-demo
runs-on: amz2023.linux.4xlarge
runs-on: amz2023.amz2023.linux.4xlarge
env:
ANDROID_NDK_VERSION: r26c
API_LEVEL: 34
Expand Down Expand Up @@ -161,7 +161,7 @@ jobs:
tokenizer: [bpe]
with:
device-type: android
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
test-infra-ref: ''
# This is the ARN of ExecuTorch project on AWS
project-arn: arn:aws:devicefarm:us-west-2:308535385114:project:02a2cf0f-6d9b-45ee-ba1a-a086587469e6
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/doc-build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ jobs:
- build-tool: buck2
with:
job-name: Build doc
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-clang12
submodules: 'true'
repository: pytorch/executorch
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/docker-builds.yml
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ concurrency:

jobs:
docker-build:
runs-on: [self-hosted, linux.2xlarge]
runs-on: [self-hosted, amz2023.linux.2xlarge]
timeout-minutes: 240
strategy:
fail-fast: false
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/lint.yml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ jobs:
lintrunner:
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-linter
fetch-depth: 0
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand Down
22 changes: 11 additions & 11 deletions .github/workflows/pull.yml
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ jobs:
- build-tool: cmake
fail-fast: false
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-gcc9
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand Down Expand Up @@ -93,7 +93,7 @@ jobs:
mode: [portable, xnnpack+custom, xnnpack+custom+qe]
fail-fast: false
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-clang12
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand Down Expand Up @@ -123,7 +123,7 @@ jobs:
- build-tool: cmake
fail-fast: false
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-clang12-android
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand All @@ -147,7 +147,7 @@ jobs:
- build-tool: cmake
fail-fast: false
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-clang12
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand All @@ -172,7 +172,7 @@ jobs:
- build-tool: cmake
fail-fast: false
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-clang12
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand All @@ -193,7 +193,7 @@ jobs:
strategy:
fail-fast: false
with:
runner: linux.12xlarge
runner: amz2023.linux.12xlarge
docker-image: executorch-ubuntu-22.04-clang12
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand Down Expand Up @@ -230,7 +230,7 @@ jobs:
- build-tool: cmake
fail-fast: false
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-clang12
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand All @@ -253,7 +253,7 @@ jobs:
- build-tool: cmake
fail-fast: false
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-clang12
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand All @@ -279,7 +279,7 @@ jobs:
strategy:
fail-fast: false
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-gcc9
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand Down Expand Up @@ -311,7 +311,7 @@ jobs:
strategy:
fail-fast: false
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-clang12
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand Down Expand Up @@ -349,7 +349,7 @@ jobs:
include:
- build-tool: buck2
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-arm-sdk
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand Down
6 changes: 3 additions & 3 deletions .github/workflows/trunk.yml
Original file line number Diff line number Diff line change
Expand Up @@ -114,7 +114,7 @@ jobs:
- build-tool: cmake
fail-fast: false
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-clang12
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand All @@ -132,7 +132,7 @@ jobs:
name: test-arm-backend-delegation
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-arm-sdk
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand All @@ -158,7 +158,7 @@ jobs:
name: test-arm-reference-delegation
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with:
runner: linux.2xlarge
runner: amz2023.linux.2xlarge
docker-image: executorch-ubuntu-22.04-arm-sdk
submodules: 'true'
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
Expand Down
Loading

0 comments on commit 7d1746b

Please sign in to comment.