set openblas env to avoid threads conflicts

7 years ago · cf9e09b115
parent 47b089a56b
commit cf9e09b115
1 changed files with 1 additions and 0 deletions
--- a/benchmark/paddle/image/run_openblas_train.sh
+++ b/benchmark/paddle/image/run_openblas_train.sh
@ -2,6 +2,7 @@ set -e

 function train() {
  unset OMP_NUM_THREADS MKL_NUM_THREADS OMP_DYNAMIC KMP_AFFINITY
+  export OPENBLAS_NUM_THREADS=1
  topology=$1
  layer_num=$2
  bs=$3