run_platform.py PMU configuration

Enable configuration of up to 4 PMU counters

Jira: MLBEDSW-4744
Change-Id: I08e9073d827033b3340bffd680288b1a83833d64

1 files changed, 12 insertions, 4 deletions

diff --git a/scripts/run_platform.py b/scripts/run_platform.py
index 93ba7cd..c0ee822 100755
--- a/scripts/run_platform.py
+++ b/scripts/run_platform.py
@@ -39,13 +39,21 @@ def run_cmd(cmd, **kwargs):
     print(f"Running command: {cmd_str}")
     return subprocess.run(cmd, check=True, **kwargs)
 
-def build_core_platform(output_folder, target, toolchain):
+
+def build_core_platform(output_folder, target, toolchain, pmu):
     build_folder = output_folder/"model"/"build"
+    inner_pmu = [0, 0, 0, 0]
+    for idx in range(len(pmu)):
+        inner_pmu[idx] = pmu[idx]
     cmake_cmd = ["cmake",
                  CORE_PLATFORM_PATH/"targets"/target,
                  f"-B{build_folder}",
                  f"-DCMAKE_TOOLCHAIN_FILE={CORE_PLATFORM_PATH/'cmake'/'toolchain'/(toolchain + '.cmake')}",
-                 f"-DBAREMETAL_PATH={output_folder}"]
+                 f"-DBAREMETAL_PATH={output_folder}",
+                 f"-DETHOSU_PMU_EVENT_0={inner_pmu[0]}",
+                 f"-DETHOSU_PMU_EVENT_1={inner_pmu[1]}",
+                 f"-DETHOSU_PMU_EVENT_2={inner_pmu[2]}",
+                 f"-DETHOSU_PMU_EVENT_3={inner_pmu[3]}"]
 
     run_cmd(cmake_cmd)
 
@@ -137,6 +145,7 @@ def main():
     }
     parser = argparse.ArgumentParser()
     parser.add_argument("-o", "--output-folder", type=pathlib.Path, default="output", help="Output folder for build and generated files")
+    parser.add_argument("--pmu", type=int, action='append', help="PMU Event Counters")
     parser.add_argument("--network-path", type=pathlib.Path, required=True, help="Path to .tflite file")
     parser.add_argument("--target", choices=target_mapping, default="corstone-300", help=f"Configure target")
     parser.add_argument("--toolchain", choices=["armclang", "arm-none-eabi-gcc"], default="armclang", help=f"Configure toolchain")
@@ -144,13 +153,12 @@ def main():
     parser.add_argument("--custom-output", type=pathlib.Path, help="Custom expected output data for network")
 
     args = parser.parse_args()
-
     args.output_folder.mkdir(exist_ok=True)
 
     try:
         optimize_network(args.output_folder, args.network_path, target_mapping[args.target])
         generate_reference_data(args.output_folder, args.network_path, args.custom_input, args.custom_output)
-        build_core_platform(args.output_folder, args.target, args.toolchain)
+        build_core_platform(args.output_folder, args.target, args.toolchain, args.pmu)
         run_model(args.output_folder)
     except subprocess.CalledProcessError as err:
         print(f"Command: '{err.cmd}' failed", file=sys.stderr)