59 Commits

Author SHA1 Message Date
d73df74cb2 Preliminary support for mkl based gelu. (#187)
* Preliminary support for mkl based gelu.

* Add the vectorized function for unary ops.

* Get the mkl specialized gelu to work.
2023-07-18 07:48:48 +01:00
28e1c07304 Process unary functions per block (#180)
* Process unary functions per block.

* Add some inline hints.
2023-07-17 10:22:33 +01:00
270997a055 Add the elu op. (#113) 2023-07-09 21:56:31 +01:00
3aac1047fe Sketch the conv1d op. 2023-07-04 10:52:34 +01:00
8ad47907f3 Add the kernels. 2023-06-30 10:26:56 +01:00
c9c468e1aa Use Map2 for binary ops. 2023-06-29 10:09:15 +01:00
8ad03a5fb6 Use Map1 on unary ops. 2023-06-29 09:37:38 +01:00
1ce3843cab Add the relu op. 2023-06-28 09:38:54 +01:00
d7f729fb8f Refactor the hierarchy. 2023-06-27 11:57:27 +02:00