Browse Source

Cap the number of parallel threads

tags/v0.3.27
Martin Kroeker GitHub 2 years ago
parent
commit
5da4c93ef2
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 7 deletions
  1. +12
    -7
      interface/lapack/getrf.c

+ 12
- 7
interface/lapack/getrf.c View File

@@ -95,14 +95,19 @@ int NAME(blasint *M, blasint *N, FLOAT *a, blasint *ldA, blasint *ipiv, blasint

#ifdef SMP
args.common = NULL;

#ifndef DOUBLE
if (args.m*args.n < 40000)
#else
if (args.m*args.n < 10000)
#endif
args.nthreads=1;
else
args.nthreads = num_cpu_avail(4);
int nmax = 40000;
#else
int nmax = 10000;
endif
if (args.m*args.n <nmax) {
args.nthreads = 1;
} else {
args.nthreads = num_cpu_avail(4);
if ((args.m*args.n)/args.nthreads <nmax)
args.nthreads = (args.m*args.n)/nmax;
}

if (args.nthreads == 1) {
#endif


Loading…
Cancel
Save