Change multithreading threshold(s) for SGEMV

This commit is contained in:
Martin Kroeker 2021-03-22 17:55:33 +01:00 committed by GitHub
parent 42f0201e21
commit 979c0cbba4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 2 deletions

View File

@ -219,11 +219,21 @@ void CNAME(enum CBLAS_ORDER order,
#ifdef SMP #ifdef SMP
//if ( 1L * m * n < 2304L * GEMM_MULTITHREAD_THRESHOLD * 1500 )
#ifdef SINGLE
if ( 1L * m * n < 2304L * GEMM_MULTITHREAD_THRESHOLD * 1500 )
nthreads = 1;
else if ( 1L * m * n < 2304L * GEMM_MULTITHREAD_THRESHOLD * 5000 ) {
nthreads = 2;
} else {
#else
if ( 1L * m * n < 2304L * GEMM_MULTITHREAD_THRESHOLD ) if ( 1L * m * n < 2304L * GEMM_MULTITHREAD_THRESHOLD )
nthreads = 1; nthreads = 1;
else else {
#endif
nthreads = num_cpu_avail(2); nthreads = num_cpu_avail(2);
fprintf(stderr, "m %d n %d nthreads %d\n",m,n,nthreads);
}
if (nthreads == 1) { if (nthreads == 1) {
#endif #endif