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

Merge pull request #3773 from staticfloat/sf/openblas_default_num_threads 

Add `OPENBLAS_DEFAULT_NUM_THREADS`
This commit is contained in:
Martin Kroeker 2022-10-13 14:15:14 +02:00 committed by GitHub
parent eece0dfd14 d2ce93179f
commit ab6009b0b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
driver/others/init.c
View File

@ -823,6 +823,8 @@ void gotoblas_affinity_init(void) {
if (numprocs == 0) numprocs = readenv_atoi("OMP_NUM_THREADS"); if (numprocs == 0) numprocs = readenv_atoi("OMP_NUM_THREADS");
if (numprocs == 0) numprocs = readenv_atoi("OPENBLAS_DEFAULT_NUM_THREADS");
numnodes = 1; numnodes = 1;
if (numprocs == 1) { if (numprocs == 1) {

8
driver/others/openblas_env.c
View File

@ -67,10 +67,16 @@ void openblas_read_env() {
openblas_env_thread_timeout=(unsigned int)ret; openblas_env_thread_timeout=(unsigned int)ret;
ret=0; ret=0;
if (readenv(p,"OPENBLAS_NUM_THREADS")) ret = atoi(p); if (readenv(p,"OPENBLAS_DEFAULT_NUM_THREADS")) ret = atoi(p);
if(ret<0) ret=0; if(ret<0) ret=0;
openblas_env_openblas_num_threads=ret; openblas_env_openblas_num_threads=ret;
ret=0;
if (readenv(p,"OPENBLAS_NUM_THREADS")) ret = atoi(p);
if(ret<0) ret=0;
if(ret != 0 || openblas_env_openblas_num_threads == 0)
openblas_env_openblas_num_threads=ret;
ret=0; ret=0;
if (readenv(p,"GOTO_NUM_THREADS")) ret = atoi(p); if (readenv(p,"GOTO_NUM_THREADS")) ret = atoi(p);
if(ret<0) ret=0; if(ret<0) ret=0;