adjust number of threads for sgemv and dgemv
This commit is contained in:
parent
d286daa2ba
commit
b985cea65d
|
@ -211,7 +211,18 @@ void CNAME(enum CBLAS_ORDER order,
|
|||
buffer = (FLOAT *)blas_memory_alloc(1);
|
||||
|
||||
#ifdef SMP
|
||||
nthreads = num_cpu_avail(2);
|
||||
|
||||
int nthreads_max = num_cpu_avail(2);
|
||||
int nthreads_avail = nthreads_max;
|
||||
|
||||
double MNK = (double) m * (double) n;
|
||||
if ( MNK <= (500.0 * 100.0 * (double) GEMM_MULTITHREAD_THRESHOLD) )
|
||||
nthreads_max = 1;
|
||||
|
||||
if ( nthreads_max > nthreads_avail )
|
||||
nthreads = nthreads_avail;
|
||||
else
|
||||
nthreads = nthreads_max;
|
||||
|
||||
if (nthreads == 1) {
|
||||
#endif
|
||||
|
|
Loading…
Reference in New Issue