Browse Source

avoid setting nthreads higher than available

pull/4748/head
Martin Kroeker GitHub 1 year ago
parent
commit
8e7e14b6aa
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      kernel/x86_64/tobf16.c

+ 3
- 3
kernel/x86_64/tobf16.c View File

@@ -135,7 +135,7 @@ void CNAME(BLASLONG n, FLOAT_TYPE * in, BLASLONG inc_in, bfloat16 * out, BLASLON
if (n <= 0) return;

#if defined(SMP)
int nthreads;
int nthreads = blas_cpu_number(1);
FLOAT_TYPE dummy_alpha;
FLOAT_TYPE dummy_c;
#endif
@@ -145,9 +145,9 @@ void CNAME(BLASLONG n, FLOAT_TYPE * in, BLASLONG inc_in, bfloat16 * out, BLASLON
nthreads = 1;
} else {
if (n/100000 < 100) {
nthreads = 4;
nthreads = MAX(nthreads,4);
} else {
nthreads = 16;
nthreads = MAX(nthreads,16);
}
}



Loading…
Cancel
Save