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

Merge 4b0de7690d into 6adc4b7b36

This commit is contained in:
oon3m0oo 2018-06-07 13:57:52 +00:00 committed by GitHub
parent 6adc4b7b36 4b0de7690d
commit e09fdbaa06
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
interface/gemm.c
View File

@ -411,20 +411,22 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE TransA, enum CBLAS_TRANS
mode |= (transa << BLAS_TRANSA_SHIFT);
mode |= (transb << BLAS_TRANSB_SHIFT);
nthreads_max = num_cpu_avail(3);
nthreads_avail = nthreads_max;
#ifndef COMPLEX
MNK = (double) args.m * (double) args.n * (double) args.k;
if ( MNK <= (65536.0 * (double) GEMM_MULTITHREAD_THRESHOLD) )
nthreads_max = 1;
else
nthreads_max = num_cpu_avail(3);
#else
MNK = (double) args.m * (double) args.n * (double) args.k;
if ( MNK <= (8192.0 * (double) GEMM_MULTITHREAD_THRESHOLD) )
nthreads_max = 1;
else
nthreads_max = num_cpu_avail(3);
#endif
args.common = NULL;
nthreads_avail = nthreads_max;
if ( nthreads_max > nthreads_avail )
args.nthreads = nthreads_avail;
else