Skip to content

Commit

Permalink
Fix Arm(R) Ethos(TM)-U55 NPU Demo app
Browse files Browse the repository at this point in the history
- Avoid installing TVM python and all its dependencies

Change-Id: Iaa4dd5429871f02961cac12476f9dc46091bcb34
  • Loading branch information
grant-arm committed Oct 25, 2021
1 parent 50823be commit 7061172
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
4 changes: 3 additions & 1 deletion apps/microtvm/ethosu/run_demo.sh
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,9 @@ mobilenet_url='https://storage.googleapis.com/download.tensorflow.org/models/mob
curl --retry 64 -sSL ${mobilenet_url} | gunzip | tar -xvf - ./mobilenet_v1_1.0_224_quant.tflite

# Compile model for Arm(R) Cortex(R)-M55 CPU and Ethos(TM)-U55 NPU
tvmc compile --target="ethos-u -accelerator_config=ethos-u55-256, \
# An alternative to using "python3 -m tvm.driver.tvmc" is to call
# "tvmc" directly once TVM has been pip installed.
python3 -m tvm.driver.tvmc compile --target="ethos-u -accelerator_config=ethos-u55-256, \
c -runtime=c --link-params -mcpu=cortex-m55 -executor=aot -interface-api=c -unpacked-api=1" \
--pass-config tir.disable_vectorize=1 ./mobilenet_v1_1.0_224_quant.tflite --output-format=mlf
tar -xvf module.tar
Expand Down
5 changes: 2 additions & 3 deletions tests/scripts/task_cpp_unittest.sh
Original file line number Diff line number Diff line change
Expand Up @@ -48,9 +48,8 @@ cd ../..

# Test Arm(R) Cortex(R)-M55 CPU and Ethos(TM)-U55 NPU demo app
FVP_PATH="/opt/arm/FVP_Corstone_SSE-300_Ethos-U55"
VELA_INSTALLED=$(pip3 list | grep vela)
if [ -d $FVP_PATH ] && [ -n "$VELA_INSTALLED" ]; then
sudo pip3 install -e python
VELA_INSTALLED="$(pip3 list | grep vela)"
if test -d $FVP_PATH && test -n "$VELA_INSTALLED"; then
cd apps/microtvm/ethosu
./run_demo.sh --fvp_path $FVP_PATH --cmake_path /opt/arm/cmake/bin/cmake
cd ../../..
Expand Down

0 comments on commit 7061172

Please sign in to comment.