From efa2cc36a265a53140abaa472e04cb22a9470095 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Tue, 21 Feb 2023 22:16:56 +0200 Subject: [PATCH] tests : fix cblas_sgemm call --- tests/test-mul-mat1.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test-mul-mat1.c b/tests/test-mul-mat1.c index 305accf..4172cfb 100644 --- a/tests/test-mul-mat1.c +++ b/tests/test-mul-mat1.c @@ -13,7 +13,7 @@ #include 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); } }