From b13f3aab4bd9872296deaa38035d1e0d265c6eb4 Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Mon, 23 Apr 2018 16:10:24 +0800
Subject: [PATCH] fix benchmark error for multi models

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

diff --git a/tools/mace_tools.py b/tools/mace_tools.py
index 27908389..dc43b1ce 100644
--- a/tools/mace_tools.py
+++ b/tools/mace_tools.py
@@ -369,7 +369,8 @@ def process_models(project_name, configs, embed_model_data, vlog_level,
                                          model_config["input_shapes"],
                                          input_file_list)
 
-        if FLAGS.mode == "build" or FLAGS.mode == "all":
+        if FLAGS.mode == "build" or FLAGS.mode == "benchmark" or \
+                FLAGS.mode == "all":
             sh_commands.gen_model_code(
                     "mace/codegen/models/%s" % model_name,
                     model_config["platform"],
-- 
GitLab