Browse Source

adjust number of threads for sgemv and dgemv

tags/v0.2.11^2
wernsaar 11 years ago
parent
commit
b985cea65d
1 changed files with 12 additions and 1 deletions
  1. +12
    -1
      interface/gemv.c

+ 12
- 1
interface/gemv.c View File

@@ -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…
Cancel
Save