floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Compare environment variables for NUM_THREADS against compile-time maximum

This commit is contained in:
Martin Kroeker 2020-11-17 19:21:12 +01:00 committed by GitHub
parent 4639c9ae4e
commit 1592c1f708
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
driver/others/openblas_env.c
View File

@ -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;
}