search

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

请问支持绑定到 torch 自定义算子吗？ #14

Open wyg1997 opened 9 months ago

wyg1997 commented 9 months ago

如题，README 中单独编译后，test 的输入是 numpy，是否可以直接绑定到 torch-npu 的算子中呢？