From d72f636c2edc28cf603436eadfba501319faa6ac Mon Sep 17 00:00:00 2001
From: tensor-tang <jian.j.tang@intel.com>
Date: Fri, 22 Sep 2017 11:17:44 +0800
Subject: [PATCH] add KMP setting and default test vgg19

---
 benchmark/paddle/image/run.mkldnn.sh | 4 ++++
 benchmark/paddle/image/vgg.py        | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/benchmark/paddle/image/run.mkldnn.sh b/benchmark/paddle/image/run.mkldnn.sh
index 03a87afbc30..9f762436959 100755
--- a/benchmark/paddle/image/run.mkldnn.sh
+++ b/benchmark/paddle/image/run.mkldnn.sh
@@ -1,5 +1,9 @@
 set -e
 
+unset OMP_NUM_THREADS MKL_NUM_THREADS
+export OMP_DYNAMIC="FALSE"
+export KMP_AFFINITY="granularity=fine,compact,0,0"
+
 function train() {
   topology=$1
   bs=$2
diff --git a/benchmark/paddle/image/vgg.py b/benchmark/paddle/image/vgg.py
index 69e4a4cddb4..b8429975f5c 100644
--- a/benchmark/paddle/image/vgg.py
+++ b/benchmark/paddle/image/vgg.py
@@ -5,7 +5,7 @@ height = 224
 width = 224
 num_class = 1000
 batch_size = get_config_arg('batch_size', int, 64)
-layer_num = get_config_arg('layer_num', int, 16)
+layer_num = get_config_arg('layer_num', int, 19)
 
 args = {'height': height, 'width': width, 'color': True, 'num_class': num_class}
 define_py_data_sources2(
-- 
GitLab