oon3m0oo
GitHub
8 years ago
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Unified View
Diff Options
-
+5 -3interface/gemm.c
+ 5
- 3
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 |= (transa << BLAS_TRANSA_SHIFT); | ||||
| mode |= (transb << BLAS_TRANSB_SHIFT); | mode |= (transb << BLAS_TRANSB_SHIFT); | ||||
| nthreads_max = num_cpu_avail(3); | |||||
| nthreads_avail = nthreads_max; | |||||
| #ifndef COMPLEX | #ifndef COMPLEX | ||||
| MNK = (double) args.m * (double) args.n * (double) args.k; | MNK = (double) args.m * (double) args.n * (double) args.k; | ||||
| if ( MNK <= (65536.0 * (double) GEMM_MULTITHREAD_THRESHOLD) ) | if ( MNK <= (65536.0 * (double) GEMM_MULTITHREAD_THRESHOLD) ) | ||||
| nthreads_max = 1; | nthreads_max = 1; | ||||
| else | |||||
| nthreads_max = num_cpu_avail(3); | |||||
| #else | #else | ||||
| MNK = (double) args.m * (double) args.n * (double) args.k; | MNK = (double) args.m * (double) args.n * (double) args.k; | ||||
| if ( MNK <= (8192.0 * (double) GEMM_MULTITHREAD_THRESHOLD) ) | if ( MNK <= (8192.0 * (double) GEMM_MULTITHREAD_THRESHOLD) ) | ||||
| nthreads_max = 1; | nthreads_max = 1; | ||||
| else | |||||
| nthreads_max = num_cpu_avail(3); | |||||
| #endif | #endif | ||||
| args.common = NULL; | args.common = NULL; | ||||
| nthreads_avail = nthreads_max; | |||||
| if ( nthreads_max > nthreads_avail ) | if ( nthreads_max > nthreads_avail ) | ||||
| args.nthreads = nthreads_avail; | args.nthreads = nthreads_avail; | ||||
| else | else | ||||