op fusion only, or do codegen like tvm ?

mindspore-ai / akg

AKG (Auto Kernel Generator) is an optimizer for operators in Deep Learning Networks, which provides the ability to automatically fuse ops with specific patterns.

https://gitee.com/mindspore/akg

Apache License 2.0

214 stars 38 forks source link

op fusion only, or do codegen like tvm ? #1

Closed ZongqiangZhang closed 3 years ago

ZongqiangZhang commented 4 years ago

i am not sure the roadmap of AKG. It will focus on op fusion only, or do kernel codegen just like TVM.

RyanWhb commented 4 years ago

AKG supports kernel generation for Ascend and GPU platform now，u can develop operators by using TVM DSL.

leonwanghui commented 3 years ago

@ZongqiangZhang This issue would be closed, please be free to reopen if you have any question, thanks!