We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d66d6c6 commit a3b2b7bCopy full SHA for a3b2b7b
benchmark/paddle/image/run_mkldnn.sh
@@ -1,9 +1,7 @@
1
set -e
2
3
function train() {
4
- unset OMP_NUM_THREADS MKL_NUM_THREADS
5
- export OMP_DYNAMIC="FALSE"
6
- export KMP_AFFINITY="granularity=fine,compact,0,0"
+ unset OMP_NUM_THREADS MKL_NUM_THREADS OMP_DYNAMIC KMP_AFFINITY
7
topology=$1
8
layer_num=$2
9
bs=$3
@@ -14,8 +12,6 @@ function train() {
14
12
elif [ $4 == "False" ]; then
15
13
thread=`nproc`
16
# each trainer_count use only 1 core to avoid conflict
17
- export OMP_NUM_THREADS=1
18
- export MKL_NUM_THREADS=1
19
log="logs/${topology}-${layer_num}-${thread}mklml-${bs}.log"
20
else
21
echo "Wrong input $3, use True or False."
0 commit comments