]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
tests : fix cblas_sgemm call
authorGeorgi Gerganov <redacted>
Tue, 21 Feb 2023 20:16:56 +0000 (22:16 +0200)
committerGeorgi Gerganov <redacted>
Tue, 21 Feb 2023 20:16:56 +0000 (22:16 +0200)
tests/test-mul-mat1.c

index 305accf55b359e6dc62142db3f0b9acbb2c405dd..4172cfb6b27a85e95c999c2e8516487cd1ad7ae7 100644 (file)
@@ -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);
         }
     }