Changeset 1df9cd1


Ignore:
Timestamp:
Jan 28, 2019, 10:50:54 PM (6 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:
1 edited

Legend:

Unmodified
Added
Removed
  • src/ai/tensor.c

    r21631e9 r1df9cd1  
    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.