From 2b13dd2c2411209e4a0ff33181a34ec12ff371e0 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Wed, 7 Mar 2018 16:22:36 +0800
Subject: [PATCH] Model benchmark tool support multiple inputs or outputs.

---
 mace_run.cc | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/mace_run.cc b/mace_run.cc
index 0e7f20ee..0e1a7d81 100644
--- a/mace_run.cc
+++ b/mace_run.cc
@@ -274,12 +274,12 @@ bool MultipleInputOrOutput(const std::vector<std::string> &input_names,
   LOG(INFO) << "Net init latency: " << t1 - t0 << " us";
   LOG(INFO) << "Total init latency: " << init_micros << " us";
 
-  const int input_count = input_names.size();
-  const int output_count = output_names.size();
+  const size_t input_count = input_names.size();
+  const size_t output_count = output_names.size();
   std::vector<mace::MaceInputInfo> input_infos(input_count);
   std::map<std::string, float*> outputs;
   std::vector<std::unique_ptr<float[]>> input_datas(input_count);
-  for (int i = 0; i < input_count; ++i) {
+  for (size_t i = 0; i < input_count; ++i) {
     // Allocate input and output
     int64_t input_size =
         std::accumulate(input_shapes[i].begin(), input_shapes[i].end(), 1,
@@ -300,7 +300,7 @@ bool MultipleInputOrOutput(const std::vector<std::string> &input_names,
     input_infos[i].data = input_datas[i].get();
   }
   std::vector<std::unique_ptr<float[]>> output_datas(output_count);
-  for (int i = 0; i < output_count; ++i) {
+  for (size_t i = 0; i < output_count; ++i) {
     int64_t output_size =
         std::accumulate(output_shapes[i].begin(), output_shapes[i].end(), 1,
                         std::multiplies<int64_t>());
@@ -329,7 +329,7 @@ bool MultipleInputOrOutput(const std::vector<std::string> &input_names,
     LOG(INFO) << "Averate latency: " << (t1 - t0) / FLAGS_round << " us";
   }
 
-  for (int i = 0; i < output_count; ++i) {
+  for (size_t i = 0; i < output_count; ++i) {
     std::string output_name = FLAGS_output_file + "_" + FormatName(output_names[i]);
     ofstream out_file(output_name, ios::binary);
     int64_t output_size =
@@ -370,14 +370,14 @@ int main(int argc, char **argv) {
   std::vector<std::string> input_shapes = str_util::Split(FLAGS_input_shape, ':');
   std::vector<std::string> output_shapes = str_util::Split(FLAGS_output_shape, ':');
 
-  const int input_count = input_shapes.size();
-  const int output_count = output_shapes.size();
+  const size_t input_count = input_shapes.size();
+  const size_t output_count = output_shapes.size();
   std::vector<vector<int64_t>> input_shape_vec(input_count);
   std::vector<vector<int64_t>> output_shape_vec(output_count);
-  for (int i = 0; i < input_count; ++i) {
+  for (size_t i = 0; i < input_count; ++i) {
     ParseShape(input_shapes[i], &input_shape_vec[i]);
   }
-  for (int i = 0; i < output_count; ++i) {
+  for (size_t i = 0; i < output_count; ++i) {
     ParseShape(output_shapes[i], &output_shape_vec[i]);
   }
 
-- 
GitLab