目录
Corleone

!2137 merge test into master

Merge br_schedule into master

Created-by: corleoneliu Commit-by: liuchao;corleoneliu;huawuyi;hujiahui8;super_yuziyu;sasaki;ombre_mer1;salazar111;bigprestigee1;Corleone;lihui;lihui488 Merged-by: nicholas_yhr Description: <!– Thanks for sending a pull request! Here are some tips for you:

  1. If this is your first time, please read our contributor guidelines: https://atomgit.com/mindspore/mindspore/blob/master/CONTRIBUTING.md

  2. If you want to contribute your code but don’t know who will review and merge, please add label mindspore-assistant to the pull request, we will find and do it as soon as possible.

–>

What type of PR is this?

/kind feature


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

merge akg_scheduler to master


Which issue(s) this PR fixes:

Fixes #417


Test Plan and Test result:What scenarios were tested, and what were the verification results(Function, performance, reliability, etc.)


Self-checklist:(请自检,在[ ]内打上x,我们将检视你的完成情况,否则会导致pr无法合入

  • 设计:PR对应的方案是否已经经过Maintainer评审,方案检视意见是否均已答复并完成方案修改
  • 测试:PR中的代码是否已有UT/ST测试用例进行充分的覆盖,新增测试用例是否随本PR一并上库或已经上库
  • 验证:PR描述信息中是否已包含对该PR对应的Feature、Refactor、Bugfix的预期目标达成情况的详细验证结果描述
  • 接口:是否涉及对外接口变更,相应变更已得到接口评审组织的通过,API对应的注释信息已经刷新正确
  • 文档:是否涉及官网文档修改,如果涉及请及时提交资料到Doc仓

See merge request: mindspore/akg!2137

10天前3421次提交

Auto Kernel Generator (AKG)

项目简介

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

  • AKG-AGENT: AKG-AGENT是我们最新孵化的一款大模型驱动的算子生成工具,提供了基于 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 代码仓中的各个子项目,都支持独立安装构建,请参考各子项目使用说明。

贡献

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

版本说明

版本说明详见RELEASE

许可证

Apache License 2.0

AKG SIG 交流微信群

关于

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

136.6 MB
邀请码
    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

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