bytedance / matxscript

A high-performance, extensible Python AOT compiler.
https://bytedance.github.io/matxscript/
Apache License 2.0
412 stars 39 forks source link

[IR] kernel introduce an intermediate ir #232

Closed jc-bytedance closed 1 year ago

jc-bytedance commented 1 year ago

A new intermediate IR has been introduced in kernel parser in order to keep more shape info and high level operation info before converting to matx.ir. With this new ir:

By applying this ir: