Browse Source

Handle misleading thread count when OMP_PLACES=threads

pull/2707/head
Martin Kroeker GitHub 5 years ago
parent
commit
fb5d13e37b
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 6 deletions
  1. +7
    -6
      common_thread.h

+ 7
- 6
common_thread.h View File

@@ -135,12 +135,13 @@ static __inline int num_cpu_avail(int level) {
int openmp_nthreads=0;
#endif

if (blas_cpu_number == 1

#ifdef USE_OPENMP
|| omp_in_parallel()
#endif
) return 1;
#ifndef USE_OPENMP
if ( blas_cpu_number == 1 )
#else
if ( omp_in_parallel() ||
(blas_cpu_number == 1 && !openblas_env_omp_places_threads) )
#endif
return 1;

#ifdef USE_OPENMP
openmp_nthreads=omp_get_max_threads();


Loading…
Cancel
Save