From 4756bf12f9dbe6502b3831808d7e4f217586980a Mon Sep 17 00:00:00 2001 From: Nir Ekhauz Date: Sun, 27 Jun 2021 11:20:24 +0300 Subject: run_platform.py PMU configuration Enable configuration of up to 4 PMU counters Jira: MLBEDSW-4744 Change-Id: I08e9073d827033b3340bffd680288b1a83833d64 --- scripts/run_platform.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) (limited to 'scripts') 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) -- cgit v1.2.1