extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2024-11-08 00:54:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

ggml : add alternative cblas_sgemm call

Browse Source
This commit is contained in:
Georgi Gerganov 2022-12-08 23:48:04 +02:00
parent 832b4f34c9
commit 50a061b313
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ggml.c
View File

@ -4590,13 +4590,22 @@ void ggml_compute_forward_mul_mat_f16_f32(
// }
//}
// zT = y * xT
{
#if 1
// zT = y * xT
cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasTrans,
ne11, ne01, ne10,
1.0f, y, ne10,
x, ne10,
1.0f, y, ne00,
x, ne00,
0.0f, d, ne01);
#else
// zT = (xT * y)T
cblas_sgemm(CblasColMajor, CblasTrans, CblasNoTrans,
ne01, ne11, ne10,
1.0f, x, ne00,
y, ne00,
0.0f, d, ne01);
#endif
}
}
}