From 32afbc92685e1ccc6b6eeef4aefd0b94c6b9aa33 Mon Sep 17 00:00:00 2001
From: Zingo Andersen <zingo.andersen@arm.com>
Date: Fri, 4 Oct 2024 15:50:20 +0200
Subject: [PATCH] Arm backend: Fix pte path in run.sh when using a model python
 file

This fix a problem when you use a file path as used model
e.g. --model_name=<FILE_PATH.py>
Then the script didn't look for the generate pte in the same folers as
the aot_arm_compiler.py put it.

Signed-off-by: Zingo Andersen <zingo.andersen@arm.com>
Change-Id: I1b5adf9371101a05790dc7c3396408324daffac9
---
 examples/arm/run.sh | 27 +++++++++++++++------------
 1 file changed, 15 insertions(+), 12 deletions(-)

diff --git a/examples/arm/run.sh b/examples/arm/run.sh
index bdf80029049..e4941519449 100755
--- a/examples/arm/run.sh
+++ b/examples/arm/run.sh
@@ -94,12 +94,13 @@ fi
 function generate_pte_file() {
     [[ $# -ne 2 ]] && { echo "[${FUNCNAME[0]}]" "Expecting model and model_compiler_flags flag, got, $*"; exit 1; }
     local model=${1}
+    local model_short_name=$(basename -- "${model}" ".py")
     local model_compiler_flags=${2}
 
-    local model_filename=${model}_arm_${target}.pte
+    local model_filename=${model_short_name}_arm_${target}.pte
     if [[ "${model_compiler_flags}" == *"--delegate"* ]]; then
-	# Name aligned with default aot_arm_compiler output
-        model_filename=${model}_arm_delegate_${target}.pte
+        # Name aligned with default aot_arm_compiler output
+        model_filename=${model_short_name}_arm_delegate_${target}.pte
     fi
     cd $et_root_dir
 
@@ -233,18 +234,20 @@ function run_fvp() {
             -C mps3_board.uart0.shutdown_on_eot=1               \
             -a "${elf}"                                         \
             --timelimit 120 || true # seconds
-        echo "[${FUNCNAME[0]} Simulation complete, $?"
+        echo "[${FUNCNAME[0]}] Simulation complete, $?"
     elif [[ ${target} == *"ethos-u85"*  ]]; then
-	${fvp_model}                                            \
-	    -C mps4_board.subsystem.cpu0.CFGITCMSZ=11           \
-	    -C mps4_board.subsystem.ethosu.num_macs=${num_macs} \
-	    -C mps4_board.visualisation.disable-visualisation=1 \
-	    -C vis_hdlcd.disable_visualisation=1                \
-	    -C mps4_board.telnetterminal0.start_telnet=0        \
-	    -C mps4_board.uart0.out_file='-'                    \
-	    -C mps4_board.uart0.shutdown_on_eot=1               \
+        echo "Running ${elf} for ${target} run with FVP:${fvp_model} num_macs:${num_macs}"
+    	${fvp_model}                                            \
+            -C mps4_board.subsystem.cpu0.CFGITCMSZ=11           \
+            -C mps4_board.subsystem.ethosu.num_macs=${num_macs} \
+            -C mps4_board.visualisation.disable-visualisation=1 \
+            -C vis_hdlcd.disable_visualisation=1                \
+            -C mps4_board.telnetterminal0.start_telnet=0        \
+            -C mps4_board.uart0.out_file='-'                    \
+            -C mps4_board.uart0.shutdown_on_eot=1               \
             -a "${elf}"                                         \
             --timelimit 120 || true # seconds
+        echo "[${FUNCNAME[0]}] Simulation complete, $?"
     else
         echo "Running ${elf} for ${target} is not supported"
         exit 1