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

Add lower threshold for multithreading in ?potrf and ?potri

This commit is contained in:
Martin Kroeker 2021-06-26 23:47:41 +02:00 committed by GitHub
parent 1f8bda71b9
commit 1b5620b66e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
interface/lapack/potrf.c
View File

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

3
interface/lapack/potri.c
View File

@ -121,6 +121,9 @@ int NAME(char *UPLO, blasint *N, FLOAT *a, blasint *ldA, blasint *Info){
#ifdef SMP #ifdef SMP
args.common = NULL; args.common = NULL;
if (args.n < 180)
args.nthreads = 1;
else
args.nthreads = num_cpu_avail(4); args.nthreads = num_cpu_avail(4);
if (args.nthreads == 1) { if (args.nthreads == 1) {

7
interface/lapack/zpotrf.c
View File

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

9
interface/lapack/zpotri.c
View File

@ -121,6 +121,15 @@ int NAME(char *UPLO, blasint *N, FLOAT *a, blasint *ldA, blasint *Info){
#ifdef SMP #ifdef SMP
args.nthreads = num_cpu_avail(4); args.nthreads = num_cpu_avail(4);
#ifndef DOUBLE
if (args.n < 200)
#else
if (args.n < 150)
#endif
args.nthreads=1;
else
#endif
args.nthreads = num_cpu_avail(4);
if (args.nthreads == 1) { if (args.nthreads == 1) {
#endif #endif