From 8ad093c37b10a41d997e7e98a6bf63363a44482c Mon Sep 17 00:00:00 2001 From: Isaac Hermida Date: Wed, 18 Dec 2024 07:15:50 +0100 Subject: [PATCH] imx-machine-learning: check vela return code Return an error in case any of the models could not be converted. https://onedigi.atlassian.net/browse/DEL-9424 Signed-off-by: Isaac Hermida --- .../0003-check-vela-return-code.patch | 26 +++++++++++++++++++ .../eiq-examples/eiq-examples_git.bbappend | 1 + 2 files changed, 27 insertions(+) create mode 100644 meta-digi-dey/dynamic-layers/imx-machine-learning/recipes-libraries/eiq-examples/eiq-examples/0003-check-vela-return-code.patch diff --git a/meta-digi-dey/dynamic-layers/imx-machine-learning/recipes-libraries/eiq-examples/eiq-examples/0003-check-vela-return-code.patch b/meta-digi-dey/dynamic-layers/imx-machine-learning/recipes-libraries/eiq-examples/eiq-examples/0003-check-vela-return-code.patch new file mode 100644 index 000000000..b4172821d --- /dev/null +++ b/meta-digi-dey/dynamic-layers/imx-machine-learning/recipes-libraries/eiq-examples/eiq-examples/0003-check-vela-return-code.patch @@ -0,0 +1,26 @@ +From: Isaac Hermida +Date: Tue, 17 Dec 2024 18:34:48 +0100 +Subject: [PATCH] check vela return code + +Signed-off-by: Isaac Hermida +--- + download_models.py | 6 +++++- + 1 file changed, 5 insertions(+), 1 deletion(-) + +diff --git a/download_models.py b/download_models.py +index 462f7b755eea..fc5ab234ae47 100644 +--- a/download_models.py ++++ b/download_models.py +@@ -103,7 +103,11 @@ def convert_model(model_dir, vela_dir): + if name.endswith(".tflite"): + print('Converting', name) + model = os.path.join(model_dir, name) +- os.system('vela ' + model + " --output-dir " + vela_dir) ++ ret = os.system('vela ' + model + " --output-dir " + vela_dir) ++ if ret != 0: ++ print(f"Error: Failed to convert model {name} with vela") ++ import sys ++ sys.exit(1) + + model_dir = 'models' + vela_dir = 'vela_models' diff --git a/meta-digi-dey/dynamic-layers/imx-machine-learning/recipes-libraries/eiq-examples/eiq-examples_git.bbappend b/meta-digi-dey/dynamic-layers/imx-machine-learning/recipes-libraries/eiq-examples/eiq-examples_git.bbappend index 4991514fd..6d2fbbc32 100644 --- a/meta-digi-dey/dynamic-layers/imx-machine-learning/recipes-libraries/eiq-examples/eiq-examples_git.bbappend +++ b/meta-digi-dey/dynamic-layers/imx-machine-learning/recipes-libraries/eiq-examples/eiq-examples_git.bbappend @@ -11,6 +11,7 @@ VELA_MODELS_DIR = "vela_models" SRC_URI += " \ file://0001-Customize-EiQ-demos.patch \ file://0002-improvements-capture-x-windows-and-increase-resoluti.patch \ + file://0003-check-vela-return-code.patch \ file://scripts/launch_eiq_demo.sh \ file://service/eiqdemo.service \ "