From 354c957a52c9e7f7a654e848395bb31662954dde Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Mon, 18 Mar 2019 19:10:48 +0800
Subject: [PATCH] Fix op_test_util bug when there is no operation.

---
 mace/ops/buffer_to_image_benchmark.cc |  2 --
 mace/ops/ops_test_util.cc             | 20 +++++++++++---------
 2 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/mace/ops/buffer_to_image_benchmark.cc b/mace/ops/buffer_to_image_benchmark.cc
index d4e687b8..8249c344 100644
--- a/mace/ops/buffer_to_image_benchmark.cc
+++ b/mace/ops/buffer_to_image_benchmark.cc
@@ -50,8 +50,6 @@ void FilterBufferToImage(int iters,
                    b2i_output);
   };
 
-  // Warm-up
-  net.Setup(D);
   for (int i = 0; i < 5; ++i) {
     transform_func();
   }
diff --git a/mace/ops/ops_test_util.cc b/mace/ops/ops_test_util.cc
index dec3bd50..25de146a 100644
--- a/mace/ops/ops_test_util.cc
+++ b/mace/ops/ops_test_util.cc
@@ -188,15 +188,17 @@ bool OpsTestNet::Setup(mace::DeviceType device) {
       }
     }
   }
-  auto op_def = op_defs_.back();
-  for (int i = 0; i < op_def.output_size(); ++i) {
-    ws_.RemoveTensor(op_def.output(i));
-    auto output_info = net_def.add_output_info();
-    output_info->set_name(op_def.output(i));
-    if (op_def.output_type_size() == op_def.output_size()) {
-      output_info->set_data_type(op_def.output_type(i));
-    } else {
-      output_info->set_data_type(DataType::DT_FLOAT);
+  if (!op_defs_.empty()) {
+    auto op_def = op_defs_.back();
+    for (int i = 0; i < op_def.output_size(); ++i) {
+      ws_.RemoveTensor(op_def.output(i));
+      auto output_info = net_def.add_output_info();
+      output_info->set_name(op_def.output(i));
+      if (op_def.output_type_size() == op_def.output_size()) {
+        output_info->set_data_type(op_def.output_type(i));
+      } else {
+        output_info->set_data_type(DataType::DT_FLOAT);
+      }
     }
   }
   MemoryOptimizer mem_optimizer;
-- 
GitLab