Skip to content

Commit 670002b

Browse files
StrycekSimonrobert-kalmar
authored andcommitted
NXP backend: Add aot example test for MobileNetV2
1 parent 34c1569 commit 670002b

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

.github/workflows/pull.yml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -860,8 +860,9 @@ jobs:
860860
# Run pytest
861861
PYTHON_EXECUTABLE=python bash backends/nxp/run_unittests.sh
862862
863-
# Run aot example:
864-
PYTHON_EXECUTABLE=python bash examples/nxp/run_aot_example.sh
863+
# Run aot examples:
864+
PYTHON_EXECUTABLE=python bash examples/nxp/run_aot_example.sh cifar10
865+
PYTHON_EXECUTABLE=python bash examples/nxp/run_aot_example.sh mobilenetv2
865866
866867
867868
test-vulkan-models-linux:

examples/nxp/run_aot_example.sh

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,12 @@ set -eux
77

88
SCRIPT_DIR=$(dirname $(readlink -fm $0))
99
EXECUTORCH_DIR=$(dirname $(dirname $SCRIPT_DIR))
10+
MODEL=${1:-"cifar10"}
1011

1112
cd $EXECUTORCH_DIR
1213

1314
# Run the AoT example
1415
python -m examples.nxp.aot_neutron_compile --quantize \
15-
--delegate --neutron_converter_flavor SDK_25_03 -m cifar10
16+
--delegate --neutron_converter_flavor SDK_25_03 -m ${MODEL}
1617
# verify file exists
17-
test -f cifar10_nxp_delegate.pte
18+
test -f ${MODEL}_nxp_delegate.pte

0 commit comments

Comments
 (0)