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

Handle misleading thread count when OMP_PLACES=threads

This commit is contained in:
Martin Kroeker 2020-07-07 15:30:29 +02:00 committed by GitHub
parent a83680b40b
commit fb5d13e37b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
common_thread.h
View File

@ -135,12 +135,13 @@ static __inline int num_cpu_avail(int level) {
int openmp_nthreads=0; int openmp_nthreads=0;
#endif #endif
if (blas_cpu_number == 1 #ifndef USE_OPENMP
if ( blas_cpu_number == 1 )
#ifdef USE_OPENMP #else
|| omp_in_parallel() if ( omp_in_parallel() ||
#endif (blas_cpu_number == 1 && !openblas_env_omp_places_threads) )
) return 1; #endif
return 1;
#ifdef USE_OPENMP #ifdef USE_OPENMP
openmp_nthreads=omp_get_max_threads(); openmp_nthreads=omp_get_max_threads();