From 1592c1f708e5c3274a954095714a929131da31fc Mon Sep 17 00:00:00 2001 From: Martin Kroeker Date: Tue, 17 Nov 2020 19:21:12 +0100 Subject: [PATCH] Compare environment variables for NUM_THREADS against compile-time maximum --- driver/others/openblas_env.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/driver/others/openblas_env.c b/driver/others/openblas_env.c index 64ece9515..72aed7231 100644 --- a/driver/others/openblas_env.c +++ b/driver/others/openblas_env.c @@ -67,16 +67,19 @@ void openblas_read_env() { ret=0; if (readenv(p,"OPENBLAS_NUM_THREADS")) ret = atoi(p); if(ret<0) ret=0; + if (ret > MAX_CPU_NUMBER) ret = MAX_CPU_NUMBER; openblas_env_openblas_num_threads=ret; ret=0; if (readenv(p,"GOTO_NUM_THREADS")) ret = atoi(p); if(ret<0) ret=0; + if (ret > MAX_CPU_NUMBER) ret = MAX_CPU_NUMBER; openblas_env_goto_num_threads=ret; ret=0; if (readenv(p,"OMP_NUM_THREADS")) ret = atoi(p); if(ret<0) ret=0; + if (ret > MAX_CPU_NUMBER) ret = MAX_CPU_NUMBER; openblas_env_omp_num_threads=ret; }