Merge pull request #5872 from jschueller/tcap

martin-frbg · web-flow · commit baa253650af9 · 2026-07-08T13:58:38.000+02:00
Add missing thread caps
diff --git a/interface/lapack/getrs.c b/interface/lapack/getrs.c
@@ -126,7 +126,11 @@ int NAME(char *TRANS, blasint *N, blasint *NRHS, FLOAT *a, blasint *ldA,
 
 #ifdef SMP
   args.common = NULL;
-  args.nthreads = num_cpu_avail(4);
+  if (args.m < 64) {
+    args.nthreads = 1;
+  } else {
+    args.nthreads = num_cpu_avail(4);
+  }
 
   if (args.nthreads == 1) {
 #endif
diff --git a/interface/lapack/laed3.c b/interface/lapack/laed3.c
@@ -72,7 +72,8 @@ int NAME(blasint *k, blasint *n, blasint *n1, FLOAT *d,
   if (kval == 0) return 0;
 
 #ifdef SMP
-  int nthreads = num_cpu_avail(4);
+  int nthreads = 1;
+  if (n >= 64) nthreads = num_cpu_avail(4);
 
   if (nthreads == 1) {
 #endif
diff --git a/interface/lapack/laswp.c b/interface/lapack/laswp.c
@@ -77,7 +77,11 @@ int NAME(blasint *N, FLOAT *a, blasint *LDA, blasint *K1, blasint *K2, blasint *
   flag = (incx < 0);
 
 #ifdef SMP
-  nthreads = num_cpu_avail(1);
+  if (n < 64) {
+    nthreads = 1;
+  } else {
+    nthreads = num_cpu_avail(1);
+  }
 
   if (nthreads == 1) {
 #endif
diff --git a/interface/lapack/lauum.c b/interface/lapack/lauum.c
@@ -112,7 +112,11 @@ int NAME(char *UPLO, blasint *N, FLOAT *a, blasint *ldA, blasint *Info){
 
 #ifdef SMP
   args.common = NULL;
-  args.nthreads = num_cpu_avail(4);
+  if (args.n < 64) {
+    args.nthreads = 1;
+  } else {
+    args.nthreads = num_cpu_avail(4);
+  }
 
   if (args.nthreads == 1) {
 #endif
diff --git a/interface/lapack/trtrs.c b/interface/lapack/trtrs.c
@@ -147,7 +147,11 @@ int NAME(char *UPLO, char* TRANS, char* DIAG, blasint *N, blasint *NRHS, FLOAT *
 
 #ifdef SMP
   args.common = NULL;
-  args.nthreads = num_cpu_avail(4);
+  if (args.m < 64) {
+    args.nthreads = 1;
+  } else {
+    args.nthreads = num_cpu_avail(4);
+  }
 
   if (args.nthreads == 1) {
 #endif
diff --git a/interface/lapack/zgetrs.c b/interface/lapack/zgetrs.c
@@ -125,7 +125,11 @@ int NAME(char *TRANS, blasint *N, blasint *NRHS, FLOAT *a, blasint *ldA,
 #endif
 
 #ifdef SMP
-  args.nthreads = num_cpu_avail(4);
+  if (args.m < 64) {
+    args.nthreads = 1;
+  } else {
+    args.nthreads = num_cpu_avail(4);
+  }
 
   if (args.nthreads == 1) {
 #endif
diff --git a/interface/lapack/zlaswp.c b/interface/lapack/zlaswp.c
@@ -78,7 +78,11 @@ int NAME(blasint *N, FLOAT *a, blasint *LDA, blasint *K1, blasint *K2, blasint *
   flag = (incx < 0);
 
 #ifdef SMP
-  nthreads = num_cpu_avail(2);
+  if (n < 64) {
+    nthreads = 1;
+  } else {
+    nthreads = num_cpu_avail(2);
+  }
 
   if (nthreads == 1) {
 #endif
diff --git a/interface/lapack/zlauum.c b/interface/lapack/zlauum.c
@@ -112,7 +112,11 @@ int NAME(char *UPLO, blasint *N, FLOAT *a, blasint *ldA, blasint *Info){
 
 #ifdef SMP
   args.common = NULL;
-  args.nthreads = num_cpu_avail(4);
+  if (args.n < 64) {
+    args.nthreads = 1;
+  } else {
+    args.nthreads = num_cpu_avail(4);
+  }
 
   if (args.nthreads == 1) {
 #endif
diff --git a/interface/lapack/ztrtri.c b/interface/lapack/ztrtri.c
@@ -125,7 +125,11 @@ int NAME(char *UPLO, char *DIAG, blasint *N, FLOAT *a, blasint *ldA, blasint *In
 
 #ifdef SMP
   args.common = NULL;
-  args.nthreads = num_cpu_avail(4);
+  if (args.n < 64) {
+    args.nthreads = 1;
+  } else {
+    args.nthreads = num_cpu_avail(4);
+  }
 
   if (args.nthreads == 1) {
 #endif
diff --git a/interface/lapack/ztrtrs.c b/interface/lapack/ztrtrs.c
@@ -147,7 +147,11 @@ int NAME(char *UPLO, char* TRANS, char* DIAG, blasint *N, blasint *NRHS, FLOAT *
 
 #ifdef SMP
   args.common = NULL;
-  args.nthreads = num_cpu_avail(4);
+  if (args.m < 64) {
+    args.nthreads = 1;
+  } else {
+    args.nthreads = num_cpu_avail(4);
+  }
 
   if (args.nthreads == 1) {
 #endif
diff --git a/interface/spr.c b/interface/spr.c
@@ -194,6 +194,9 @@ void CNAME(enum CBLAS_ORDER order,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/spr2.c b/interface/spr2.c
@@ -198,6 +198,9 @@ void CNAME(enum CBLAS_ORDER order,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/syr.c b/interface/syr.c
@@ -197,6 +197,9 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, blasint n, FLOAT alpha,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/syr2.c b/interface/syr2.c
@@ -199,6 +199,9 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, blasint n, FLOAT alpha,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/tbmv.c b/interface/tbmv.c
@@ -225,6 +225,9 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/tpmv.c b/interface/tpmv.c
@@ -223,6 +223,9 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/zhpmv.c b/interface/zhpmv.c
@@ -196,6 +196,9 @@ void CNAME(enum CBLAS_ORDER order,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/zspr.c b/interface/zspr.c
@@ -123,6 +123,9 @@ void NAME(char *UPLO, blasint *N, FLOAT  *ALPHA,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/zspr2.c b/interface/zspr2.c
@@ -126,6 +126,9 @@ void NAME(char *UPLO, blasint *N, FLOAT  *ALPHA,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/zsymv.c b/interface/zsymv.c
@@ -120,6 +120,9 @@ void NAME(char *UPLO, blasint *N, FLOAT  *ALPHA, FLOAT *a, blasint *LDA,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/zsyr.c b/interface/zsyr.c
@@ -208,6 +208,9 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo, int n, void* valpha, FL
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/zsyr2.c b/interface/zsyr2.c
@@ -128,6 +128,9 @@ void NAME(char *UPLO, blasint *N, FLOAT  *ALPHA,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/ztbmv.c b/interface/ztbmv.c
@@ -240,6 +240,9 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
diff --git a/interface/ztpmv.c b/interface/ztpmv.c
@@ -231,6 +231,9 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_UPLO Uplo,
 #ifdef SMP
   nthreads = num_cpu_avail(2);
 
+  if (n < 50) nthreads = 1;
+  if (nthreads > 2 && n < 500) nthreads = 2;
+
   if (nthreads == 1) {
 #endif
 
