tests : fix cblas_sgemm call

pull/26/head
Georgi Gerganov 1 year ago
parent 3b3ad42906
commit efa2cc36a2
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

@ -13,7 +13,7 @@
#include <Accelerate/Accelerate.h>
const int M = 1280;
const int N = 1500;
const int N = 1536;
const int K = 1280;
uint64_t get_time_us() {
@ -284,7 +284,7 @@ int main(int argc, const char ** argv) {
if (method == 4) {
// Use BLAS sgemm from Accelerate framework
cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, M, K, N, 1.0f, src0, N, src1, N, 0.0f, dst, N);
cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasTrans, M, N, K, 1.0f, src0, K, src1, K, 0.0f, dst, N);
}
}

Loading…
Cancel
Save