@@ -1066,3 +1066,48 @@ 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+ # The generic Linux job chooses to use base env, not the one setup by the image
1089+ CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
1090+ conda activate "${CONDA_ENV}"
1091+
1092+ source .ci/scripts/utils.sh
1093+
1094+ # Validate the Raspberry Pi setup scripts
1095+ ## Pi5
1096+ examples/raspberry_pi/setup.sh pi5
1097+ examples/raspberry_pi/setup.sh pi5 --clean
1098+ ## Pi4
1099+ examples/raspberry_pi/setup.sh pi4
1100+ examples/raspberry_pi/setup.sh pi4 --clean
1101+
1102+ # Validate the script for Pico2 setup
1103+ ## Generate model
1104+ cd ./examples/raspberry_pi/pico2
1105+ ## Creates balanced_tiny_mlp_mnist.pte
1106+ python export_mlp_mnist.py
1107+ cd -
1108+
1109+ .ci/scripts/setup-arm-baremetal-tools.sh
1110+ source examples/arm/ethos-u-scratch/setup_path.sh
1111+ examples/raspberry_pi/pico2/build_firmware_pico.sh --model=balanced_tiny_mlp_mnist.pte
1112+ examples/raspberry_pi/pico2/build_firmware_pico.sh --clean
1113+ rm ./examples/raspberry_pi/pico2/balanced_tiny_mlp_mnist.pte
0 commit comments