Skip to content

Commit afb0257

Browse files
committed
NXP backend: Add aot example test for MobileNetV2
1 parent 2e3c21b commit afb0257

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
@@ -793,8 +793,9 @@ jobs:
793793
# Run pytest
794794
PYTHON_EXECUTABLE=python bash backends/nxp/run_unittests.sh
795795
796-
# Run aot example:
797-
PYTHON_EXECUTABLE=python bash examples/nxp/run_aot_example.sh
796+
# Run aot examples:
797+
PYTHON_EXECUTABLE=python bash examples/nxp/run_aot_example.sh cifar10
798+
PYTHON_EXECUTABLE=python bash examples/nxp/run_aot_example.sh mobilenetv2
798799
799800
800801
nxp-build-test:

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)