hikettei / Caten

[wip] Deep Learning Compiler based on Polyhedral Compiler and Light-weight IRs based on Optimizing Pattern Matcher

https://hikettei.github.io/Caten/

Other

20 stars 4 forks source link

Refactor: Module supports multiple outputs + view ops #212

Closed hikettei closed 1 week ago

hikettei commented 1 week ago

210

[ ] !concatenate
[ ] !chunk
[ ] test-gemm: View with offset + !gemm
[ ] MHA Working?
[ ] :shrink merge?
[ ] update threefry2x32
[x] does model support multiple outputs btw?
[ ] this is as exceptedbut !split wont work with jit
[ ] adding tests for another pr