Changeset 1df9cd1

Timestamp:

Jan 28, 2019, 10:50:54 PM (5 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/crepe_org

Children:

4b2d174

Parents:

21631e9

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
 }