floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update tobf16.c

This commit is contained in:
Martin Kroeker 2024-06-11 15:24:33 +02:00 committed by GitHub
parent 8e7e14b6aa
commit 242026cede
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
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 (n <= 0) return;
#if defined(SMP) #if defined(SMP)
int nthreads = blas_cpu_number(1); int nthreads;
FLOAT_TYPE dummy_alpha; FLOAT_TYPE dummy_alpha;
FLOAT_TYPE dummy_c; FLOAT_TYPE dummy_c;
#endif #endif
@ -144,10 +144,11 @@ void CNAME(BLASLONG n, FLOAT_TYPE * in, BLASLONG inc_in, bfloat16 * out, BLASLON
if (inc_in == 0 || inc_out == 0 || n <= 100000) { if (inc_in == 0 || inc_out == 0 || n <= 100000) {
nthreads = 1; nthreads = 1;
} else { } else {
nthreads = num_cpu_avail(1);
if (n/100000 < 100) { if (n/100000 < 100) {
nthreads = MAX(nthreads,4); nthreads = MAX(nthreads,4);
} else { // } else {
nthreads = MAX(nthreads,16); // nthreads = MAX(nthreads,16);
} }
} }