提交 aaae5f63 编写于 作者: Y yejianwu

fix benchmark model data upload to falcon

上级 92ada581
...@@ -108,11 +108,12 @@ def gen_opencl_and_tuning_code(target_soc, ...@@ -108,11 +108,12 @@ def gen_opencl_and_tuning_code(target_soc,
def model_benchmark_stdout_processor(stdout, def model_benchmark_stdout_processor(stdout,
target_soc, target_soc,
abi, abi,
model_name,
runtime, runtime,
running_round, running_round,
tuning): tuning):
metrics = {} metrics = {}
for line in stdout: for line in stdout.split("\n"):
if "Aborted" in line: if "Aborted" in line:
raise Exception("Command failed") raise Exception("Command failed")
line = line.strip() line = line.strip()
...@@ -178,6 +179,7 @@ def tuning_run(runtime, ...@@ -178,6 +179,7 @@ def tuning_run(runtime,
model_benchmark_stdout_processor(stdout, model_benchmark_stdout_processor(stdout,
target_soc, target_soc,
target_abi, target_abi,
model_name,
runtime, runtime,
running_round, running_round,
tuning) tuning)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册