Merge pull request #4441 from tensor-tang/develop

should reset env every time
PaddlePaddle · Sep 28, 2017 · 4fcced9 · 4fcced9
2 parents 424c295 + cfa86a3
commit 4fcced9
Showing 1 changed file with 3 additions and 4 deletions.
diff --git 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