Browse Source

Cap the number of parallel threads

tags/v0.3.27
Martin Kroeker GitHub 1 year ago
parent
commit
a3354a7630
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      interface/gemm.c

+ 5
- 1
interface/gemm.c View File

@@ -533,8 +533,12 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS
MNK = (double) args.m * (double) args.n * (double) args.k;
if ( MNK <= (SMP_THRESHOLD_MIN * (double) GEMM_MULTITHREAD_THRESHOLD) )
args.nthreads = 1;
else
else {
args.nthreads = num_cpu_avail(3);
if (MNK/args.nthreads < SMP_THRESHOLD_MIN*(double)GEMM_MULTITHREAD_THRESHOLD)
args.nthreads = MNK/(SMP_THRESHOLD_MIN*(double)GEMM_MULTITHREAD_THRESHOLD);
}

args.common = NULL;

if (args.nthreads == 1) {


Loading…
Cancel
Save