Browse Source

Cap the number of parallel threads

tags/v0.3.27
Martin Kroeker GitHub 2 years ago
parent
commit
496106642f
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 6 deletions
  1. +10
    -6
      interface/lapack/potrf.c

+ 10
- 6
interface/lapack/potrf.c View File

@@ -113,13 +113,17 @@ int NAME(char *UPLO, blasint *N, FLOAT *a, blasint *ldA, blasint *Info){
#ifdef SMP
args.common = NULL;
#ifndef DOUBLE
if (args.n <128)
#else
if (args.n <64)
#endif
int nmax = 128;
#else
int nmax = 64;
endif
if (args.n <nmax) {
args.nthreads = 1;
else
args.nthreads = num_cpu_avail(4);
} else {
args.nthreads = num_cpu_avail(4);
if (args.n/args.nthreads <nmax)
args.nthreads = args.n/nmax;
}

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


Loading…
Cancel
Save