fix blas matmul function

This commit is contained in:
Abhilash Majumder 2024-01-23 17:56:37 +05:30 committed by GitHub
parent b42a32d31a
commit 5f83a12382
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

2
ggml.c
View file

@ -9958,7 +9958,7 @@ static void ggml_compute_forward_mul_mat(
#endif
#if defined(GGML_USE_ACCELERATE) || defined(GGML_USE_OPENBLAS)
if (ggml_compute_forward_mul_mat_use_blas(dst)) {
if (ggml_compute_forward_mul_mat_use_blas(src0, src1, dst)) {
const int64_t ne_plane = ne01*ne00;
const int64_t desired_wsize = ne13*ne12*ne_plane*sizeof(float);
UNUSED(desired_wsize);