Changeset 461a89f for src

Timestamp:

Dec 29, 2021, 5:52:00 PM (3 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe

Children:

17189b2

Parents:

e7779f0

git-author:

Paul Brossier <piem@piem.org> (01/28/19 22:50:54)

git-committer:

Paul Brossier <piem@piem.org> (12/29/21 17:52:00)

Message:

[tensor] use cblas alias in matmul, allow n-dimensional tensors

File:

• src/ai/tensor.c (modified) (1 diff)

src/ai/tensor.c

@@ -229 +229 @@
   }
 #else
-  cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, a->shape[0],
-      b->shape[1], b->shape[0], 1.F, a->buffer, a->shape[1], b->buffer,
-      b->shape[1], 0.F, c->buffer, b->shape[1]);
+  aubio_cblas__gemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, a->shape[0],
+      b->size/b->shape[0], b->shape[0], 1.F, a->buffer, a->size/a->shape[0],
+      b->buffer,
+      b->size/b->shape[0], 0.F, c->buffer, b->size/b->shape[0]);
 #endif
 }