From cfa86a3f70cb5f2517a802f32f2c88d48ab4e0e0 Mon Sep 17 00:00:00 2001
From: tensor-tang <jian.j.tang@intel.com>
Date: Wed, 27 Sep 2017 22:10:21 +0800
Subject: [PATCH] should reset env every time

---
 benchmark/paddle/image/run_mkldnn.sh | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/benchmark/paddle/image/run_mkldnn.sh b/benchmark/paddle/image/run_mkldnn.sh
index 81de1a0e910..e31fec1cd85 100755
--- a/benchmark/paddle/image/run_mkldnn.sh
+++ b/benchmark/paddle/image/run_mkldnn.sh
@@ -1,10 +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() {
+  unset OMP_NUM_THREADS MKL_NUM_THREADS
+  export OMP_DYNAMIC="FALSE"
+  export KMP_AFFINITY="granularity=fine,compact,0,0"
   topology=$1
   bs=$2
   use_mkldnn=$3
-- 
GitLab