diff --git a/tools/mace_tools.py b/tools/mace_tools.py index 86071dbb9639d891712b38493ea2425d309d60fd..27908389d8c8c50a44d8de6c74b72914a90beea4 100644 --- a/tools/mace_tools.py +++ b/tools/mace_tools.py @@ -108,11 +108,12 @@ def gen_opencl_and_tuning_code(target_soc, def model_benchmark_stdout_processor(stdout, target_soc, abi, + model_name, runtime, running_round, tuning): metrics = {} - for line in stdout: + for line in stdout.split("\n"): if "Aborted" in line: raise Exception("Command failed") line = line.strip() @@ -178,6 +179,7 @@ def tuning_run(runtime, model_benchmark_stdout_processor(stdout, target_soc, target_abi, + model_name, runtime, running_round, tuning)