目录
master
分支21
标签12
+ 疑修
Web IDE
liuruotao
!1869 merge master into master


update pr template


Created-by: lllrt
Commit-by: lllrt
Merged-by: zichun_ye
Description: <!–  Thanks for sending a pull request!  Here are some tips for you:


If this is your first time, please read our contributor guidelines: https://github.com/mindspore-ai/mindspore/blob/master/CONTRIBUTING.md
–>


What type of PR is this?




Uncomment only one  /kind <> line, hit enter to put that in a new line, and remove leading whitespaces from that line:


/kind bug

/kind task

/kind feature




What does this PR do / why do we need it:


Which issue(s) this PR fixes:



Fixes #


Special notes for your reviewers:


See merge request: mindspore/akg!1869
3天前2341次提交
README.md

Auto Kernel Generator (AKG)

项目简介

AKG(Auto Kernel Generator)是一个深度学习编译器，用以对深度神经网络中的算子进行优化，并提供特定模式下的算子自动融合功能。 从 2.4 版本之后，我们对 AKG 进行了全新升级，当前的 AKG 项目中，包含了 AIKG、AKG-MLIR 2个子项目，支持 CPU、NVIDIA、V100/A100、Atlas 800T A2/A3、300I DUO 等多硬件后端的算子自动生成。

  • AIKG： AIKG(AI Driven Kernel Generator)是我们最新孵化的一款大模型驱动的算子生成工具，提供了基于 Multi-agent 的算子自动生成框架、AscendKernelBench 评测平台、以及算子知识库。当前支持通过 LLM 的方式，生成 Triton-Ascend（Atalas A2/A3）以及 SWFT（300I DUO）的 DSL，并完成算子 kernel 编译。
  • AKG-MLIR： 基于 MLIR 开源项目演进的深度学习编译器，提供了 CPU/GPU/Ascend 上完整的算子编译 Pipeline。当前包含了 MindSpore Dialect 图编译方言，并对 Linalg、Affine、GPU 等方言进行了扩展，增强了循环融合调度能力。此外，AKG-MLIR 对接了 AscendNPU IR，支持昇腾后端融合算子生成。

安装及使用说明

AKG 代码仓中的各个子项目，都支持独立安装构建，请参考各子项目使用说明。

  • AIKG： 请参考 AIKG 文档
  • AKG-MLIR： 请参考 AKG-MLIR 文档

贡献

欢迎您的贡献，具体细节请参考MindSpore 贡献者 Wiki。

版本说明

版本说明详见RELEASE。

许可证

Apache License 2.0。

AKG SIG 交流微信群

关于

AKG(Auto Kernel Generator)是一个深度学习编译器，用以对深度神经网络中的算子进行优化，并提供特定模式下的算子自动融合功能。

pythonc++mlir
README.md
111.7 MB
邀请码
    Gitlink（确实开源）
  • 加入我们
  • 官网邮箱：gitlink@ccf.org.cn
    • QQ群
  • QQ群
    • 公众号
  • 公众号

版权所有：中国计算机学会技术支持：开源发展技术委员会
京ICP备13000930号-9 京公网安备 11010802032778号