From 2dff98ca11a48afcceedbfb4ec6ead4eddff0118 Mon Sep 17 00:00:00 2001 From: tensor-tang 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 4d1d3e1b564..a4527e04968 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