set openblas env to avoid threads conflicts

cf9e09b1 · tensor-tang · 47b089a5 · cf9e09b1
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

benchmark/paddle/image/run_openblas_train.sh benchmark/paddle/image/run_openblas_train.sh +1 -0

未找到文件。
--- 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