Skip to content

Commit 1983436

Browse files
authored
trunk.yml: Add CI job to validate RPI tutorial scripts
1 parent 5914f8f commit 1983436

File tree

1 file changed

+47
-0
lines changed

1 file changed

+47
-0
lines changed

.github/workflows/trunk.yml

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1066,3 +1066,50 @@ jobs:
10661066
10671067
.ci/scripts/test_model.ps1 -modelName ${{ matrix.model }} -backend ${{ matrix.backend }}
10681068
}"
1069+
1070+
test-rpi-scripts:
1071+
name: test-rpi-scripts
1072+
uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@release/2.9
1073+
strategy:
1074+
matrix:
1075+
include:
1076+
- build-tool: cmake
1077+
fail-fast: false
1078+
permissions:
1079+
id-token: write
1080+
contents: read
1081+
with:
1082+
runner: linux.2xlarge
1083+
docker-image: ci-image:executorch-ubuntu-22.04-arm-sdk
1084+
submodules: 'recursive'
1085+
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
1086+
timeout: 90
1087+
script: |
1088+
1089+
# The generic Linux job chooses to use base env, not the one setup by the image
1090+
CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
1091+
conda activate "${CONDA_ENV}"
1092+
1093+
# Validate the Raspberry Pi setup scripts
1094+
1095+
## Pi5
1096+
examples/raspberry_pi/setup.sh pi5
1097+
examples/raspberry_pi/setup.sh pi5 --clean
1098+
1099+
## Pi4
1100+
1101+
examples/raspberry_pi/setup.sh pi4
1102+
examples/raspberry_pi/setup.sh pi4 --clean
1103+
1104+
# Validate the script for Pico2 setup
1105+
1106+
## Generate model
1107+
cd ./examples/raspberry_pi/pico2
1108+
## Creates balanced_tiny_mlp_mnist.pte
1109+
python export_mlp_mnist.py
1110+
cd -
1111+
examples/raspberry_pi/pico2/build_firmware_pico.sh --model=balanced_tiny_mlp_mnist.pte
1112+
1113+
examples/raspberry_pi/pico2/build_firmware_pico.sh --clean
1114+
rm ./examples/raspberry_pi/pico2/balanced_tiny_mlp_mnist.pte
1115+

0 commit comments

Comments
 (0)