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 \ "