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

Cap the number of parallel threads

This commit is contained in:
Martin Kroeker 2024-03-27 20:34:55 +01:00 committed by GitHub
parent 496106642f
commit 5da4c93ef2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
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 #ifdef SMP
args.common = NULL; args.common = NULL;
#ifndef DOUBLE #ifndef DOUBLE
if (args.m*args.n < 40000) int nmax = 40000;
#else #else
if (args.m*args.n < 10000) int nmax = 10000;
#endif endif
args.nthreads=1; if (args.m*args.n <nmax) {
else args.nthreads = 1;
} else {
args.nthreads = num_cpu_avail(4); 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) { if (args.nthreads == 1) {
#endif #endif