From 2dff98ca11a48afcceedbfb4ec6ead4eddff0118 Mon Sep 17 00:00:00 2001
From: tensor-tang <jian.j.tang@intel.com>
Date: Tue, 7 Nov 2017 23:03:11 +0800
Subject: [PATCH] remove auto setting from HT, since it's hard to unify with
 MacOS

---
 benchmark/paddle/image/run_mkldnn.sh | 13 ++-----------
 1 file changed, 2 insertions(+), 11 deletions(-)

diff --git a/benchmark/paddle/image/run_mkldnn.sh b/benchmark/paddle/image/run_mkldnn.sh
index 4d1d3e1b56..a4527e0496 100755
--- a/benchmark/paddle/image/run_mkldnn.sh
+++ b/benchmark/paddle/image/run_mkldnn.sh
@@ -2,6 +2,8 @@ set -e
 
 function train() {
   unset OMP_NUM_THREADS MKL_NUM_THREADS
+  export OMP_DYNAMIC="FALSE"
+  export KMP_AFFINITY="granularity=fine,compact,0,0"
   topology=$1
   layer_num=$2
   bs=$3
@@ -39,17 +41,6 @@ if [ ! -d "logs" ]; then
   mkdir logs
 fi
 
-total_cores=`ls -l /sys/devices/system/cpu/ | grep "cpu[0-9]*$" | wc -l`
-online_cores=`cat /sys/devices/system/cpu/cpu*/online | grep -o '1' | wc -l`
-if [ $online_cores -eq $total_cores ]; then
-  echo "Hyper Threading is ON"
-  export KMP_AFFINITY="granularity=fine,compact,1,0"
-else
-  echo "Hyper Threading is OFF"
-  export OMP_DYNAMIC="FALSE"
-  export KMP_AFFINITY="granularity=fine,compact,0,0"
-fi
-
 for use_mkldnn in True False; do
   for batchsize in 64 128 256; do
     train vgg 19 $batchsize $use_mkldnn
-- 
GitLab