From aaae5f63a8f18af019cdb35a99ab30bde683d6fa Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Mon, 23 Apr 2018 14:56:03 +0800
Subject: [PATCH] fix benchmark model data upload to falcon

---
 tools/mace_tools.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tools/mace_tools.py b/tools/mace_tools.py
index 86071dbb..27908389 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)
-- 
GitLab