Changeset 461a89f for src


Ignore:
Timestamp:
Dec 29, 2021, 5:52:00 PM (2 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:
1 edited

Legend:

Unmodified
Added
Removed
  • src/ai/tensor.c

    re7779f0 r461a89f  
    229229  }
    230230#else
    231   cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, a->shape[0],
    232       b->shape[1], b->shape[0], 1.F, a->buffer, a->shape[1], b->buffer,
    233       b->shape[1], 0.F, c->buffer, b->shape[1]);
     231  aubio_cblas__gemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, a->shape[0],
     232      b->size/b->shape[0], b->shape[0], 1.F, a->buffer, a->size/a->shape[0],
     233      b->buffer,
     234      b->size/b->shape[0], 0.F, c->buffer, b->size/b->shape[0]);
    234235#endif
    235236}
Note: See TracChangeset for help on using the changeset viewer.