提交 a6f5f6ef 编写于 作者: T tensor-tang

at least set threads number as 1, in case trainers number is larger than processors

上级 4fbba656
......@@ -79,7 +79,10 @@ function threads_config() {
if [ -z $trainers ]; then
trainers=1
fi
threads=$((processors / trainers))
threads=$((processors / trainers))
if [ $threads -eq 0 ]; then
threads=1
fi
if [ -z "$OMP_NUM_THREADS" ]; then
export OMP_NUM_THREADS=$threads
fi
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册