From 2dabadd17a2f6b3675714bc50eed8fe54918fc32 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Fri, 8 Sep 2017 16:46:05 +0800
Subject: [PATCH] Use iters at relu and addn benchmark.

---
 mace/kernels/benchmark/addn_benchmark.cc | 8 ++++++--
 mace/kernels/benchmark/relu_benchmark.cc | 8 ++++++--
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/mace/kernels/benchmark/addn_benchmark.cc b/mace/kernels/benchmark/addn_benchmark.cc
index e735965a..d717c44e 100644
--- a/mace/kernels/benchmark/addn_benchmark.cc
+++ b/mace/kernels/benchmark/addn_benchmark.cc
@@ -32,10 +32,14 @@ static void AddNBenchmark(int iters, int n, int type) {
 
   if (type == DeviceType::CPU) {
     AddNFunctor<DeviceType::CPU, float> addn_functor;
-    addn_functor(inputs, &output[0], n);
+    while (--iters) {
+      addn_functor(inputs, &output[0], n);
+    }
   } else if (type == DeviceType::NEON) {
     AddNFunctor<DeviceType::NEON, float> neon_addn_functor;
-    neon_addn_functor(inputs, &output[0], n);
+    while (--iters) {
+      neon_addn_functor(inputs, &output[0], n);
+    }
   }
 }
 
diff --git a/mace/kernels/benchmark/relu_benchmark.cc b/mace/kernels/benchmark/relu_benchmark.cc
index c384e882..2d5067b9 100644
--- a/mace/kernels/benchmark/relu_benchmark.cc
+++ b/mace/kernels/benchmark/relu_benchmark.cc
@@ -27,10 +27,14 @@ static void ReluBenchmark(int iters, int n, int type) {
 
   if (type == DeviceType::CPU) {
     ReluFunctor<DeviceType::CPU, float> relu_functor;
-    relu_functor(&input[0], &output[0], n);
+    while (--iters) {
+      relu_functor(&input[0], &output[0], n);
+    }
   } else if (type == DeviceType::NEON) {
     ReluFunctor<DeviceType::NEON, float> neon_relu_functor;
-    neon_relu_functor(&input[0], &output[0], n);
+    while (--iters) {
+      neon_relu_functor(&input[0], &output[0], n);
+    }
   }
 }
 
-- 
GitLab