Skip to content

Commit ae1f098

Browse files
kirklandsignfacebook-github-bot
authored andcommitted
Re-enable CI with smaller model for validation (#4304)
Summary: Use a stories model for E2E validation Pull Request resolved: #4304 Reviewed By: huydhn Differential Revision: D60117299 Pulled By: kirklandsign fbshipit-source-id: e28f79f9fca310e17bb6fcf5d99a0317d778eb9c
1 parent 78df332 commit ae1f098

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

.github/workflows/android.yml

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,3 +76,34 @@ jobs:
7676
retention-days: 14
7777
if-no-files-found: ignore
7878
path: ${{ runner.temp }}/artifacts/
79+
80+
# Let's see how expensive this job is, we might want to tone it down by running it periodically
81+
test-llama-app:
82+
needs: upload-artifacts
83+
permissions:
84+
id-token: write
85+
contents: read
86+
uses: pytorch/test-infra/.github/workflows/mobile_job.yml@main
87+
strategy:
88+
matrix:
89+
# https://github.com/pytorch/executorch/blob/main/examples/demo-apps/android/LlamaDemo/README.md#alternative-2-build-from-local-machine
90+
# mentions that tiktoken is only for Llama3. So, we can export it later in another archive
91+
# like https://ossci-assets.s3.amazonaws.com/executorch-android-llama2-7b-0717.zip when this is
92+
# updated to run Llama3
93+
tokenizer: [bpe]
94+
with:
95+
device-type: android
96+
runner: linux.2xlarge
97+
test-infra-ref: ''
98+
# This is the ARN of ExecuTorch project on AWS
99+
project-arn: arn:aws:devicefarm:us-west-2:308535385114:project:02a2cf0f-6d9b-45ee-ba1a-a086587469e6
100+
# This is the custom Android device pool that only includes Samsung Galaxy S2x
101+
device-pool-arn: arn:aws:devicefarm:us-west-2:308535385114:devicepool:02a2cf0f-6d9b-45ee-ba1a-a086587469e6/e59f866a-30aa-4aa1-87b7-4510e5820dfa
102+
# Uploaded to S3 from the previous job, the name of the app comes from the project itself
103+
android-app-archive: https://gha-artifacts.s3.amazonaws.com/${{ github.repository }}/${{ github.run_id }}/artifact/llm_demo_${{ matrix.tokenizer }}/app-debug.apk
104+
android-test-archive: https://gha-artifacts.s3.amazonaws.com/${{ github.repository }}/${{ github.run_id }}/artifact/llm_demo_${{ matrix.tokenizer }}/app-debug-androidTest.apk
105+
# The test spec can be downloaded from https://ossci-assets.s3.amazonaws.com/android-llama2-device-farm-test-spec.yml
106+
test-spec: arn:aws:devicefarm:us-west-2:308535385114:upload:02a2cf0f-6d9b-45ee-ba1a-a086587469e6/abd86868-fa63-467e-a5c7-218194665a77
107+
# Among the input, this is the biggest file, so it is cached on AWS to make the test faster. Note that the file is deleted by AWS after 30
108+
# days and the job will automatically re-upload the file when that happens.
109+
extra-data: https://ossci-assets.s3.amazonaws.com/executorch-android-llama2-7b-0717.zip

0 commit comments

Comments
 (0)