From 86b8bdc0af01960eaded403ade4214faebe5c475 Mon Sep 17 00:00:00 2001
From: tensor-tang <jian.j.tang@intel.com>
Date: Thu, 21 Dec 2017 23:54:38 +0800
Subject: [PATCH] enable inference alexnet benchmark

---
 benchmark/paddle/image/alexnet.py            | 11 ++++++++++-
 benchmark/paddle/image/run_mkl_infer.sh      |  1 +
 benchmark/paddle/image/run_openblas_infer.sh |  1 +
 3 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/benchmark/paddle/image/alexnet.py b/benchmark/paddle/image/alexnet.py
index 10db1944856..b0beef8ca71 100644
--- a/benchmark/paddle/image/alexnet.py
+++ b/benchmark/paddle/image/alexnet.py
@@ -7,8 +7,17 @@ width = 227
 num_class = 1000
 batch_size = get_config_arg('batch_size', int, 128)
 use_mkldnn = get_config_arg('use_mkldnn', bool, False)
+is_infer = get_config_arg("is_infer", bool, False)
+num_samples = get_config_arg('num_samples', int, 2560)
 
-args = {'height': height, 'width': width, 'color': True, 'num_class': num_class}
+args = {
+    'height': height,
+    'width': width,
+    'color': True,
+    'num_class': num_class,
+    'is_infer': is_infer,
+    'num_samples': num_samples
+}
 define_py_data_sources2(
     "train.list", None, module="provider", obj="process", args=args)
 
diff --git a/benchmark/paddle/image/run_mkl_infer.sh b/benchmark/paddle/image/run_mkl_infer.sh
index d795bcab1b7..00942e32a55 100755
--- a/benchmark/paddle/image/run_mkl_infer.sh
+++ b/benchmark/paddle/image/run_mkl_infer.sh
@@ -79,6 +79,7 @@ fi
 # inference benchmark
 for use_mkldnn in True False; do
   for batchsize in 1 2 4 8 16; do
+    infer alexnet group2 $batchsize $use_mkldnn
     infer googlenet v1 $batchsize $use_mkldnn
     infer resnet 50 $batchsize $use_mkldnn
     infer vgg 19 $batchsize $use_mkldnn
diff --git a/benchmark/paddle/image/run_openblas_infer.sh b/benchmark/paddle/image/run_openblas_infer.sh
index c1001d3a7c9..3dad42ee0df 100755
--- a/benchmark/paddle/image/run_openblas_infer.sh
+++ b/benchmark/paddle/image/run_openblas_infer.sh
@@ -56,6 +56,7 @@ fi
 
 # inference benchmark
 for batchsize in 1 2 4 8 16; do
+  infer alexnet group2 $batchsize $use_mkldnn
   infer googlenet v1 $batchsize
   infer resnet 50 $batchsize
   infer vgg 19 $batchsize
-- 
GitLab