Skip to content

Commit

Permalink
tests : fix cblas_sgemm call
Browse files Browse the repository at this point in the history
  • Loading branch information
ggerganov committed Feb 21, 2023
1 parent 3b3ad42 commit efa2cc3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test-mul-mat1.c
Original file line number Diff line number Diff line change
Expand Up @@ -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() {
Expand Down Expand Up @@ -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);
}
}

Expand Down

0 comments on commit efa2cc3

Please sign in to comment.