ggml : add alternative cblas_sgemm call

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

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, cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasTrans,
ne11, ne01, ne10, ne11, ne01, ne10,
1.0f, y, ne10, 1.0f, y, ne00,
x, ne10, x, ne00,
0.0f, d, ne01); 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
} }
} }
} }