昇腾社区首页
中文
注册
算子开发
算子开发是针对神经网络基础单元(即算子)的设计与实现过程,其目的是让开发者提出的创新算法能够在昇腾AI处理器上高效运行。面向算子开发场景,CANN提供以下多种算子开发方式,助力AI开发者低成本完成算子开发和模型调优部署:
  • 提供Ascend C编程语言,原生支持C和C++标准规范,最大化匹配用户开发习惯
  • 提供CATLASS算子模板库,通过优质代码样例与模块化组件加速矩阵类算子定制开发
  • 针对Python化编程需求,提供Triton语言支持,大幅降低算子开发门槛

Ascend C算子开发

基于 Ascend C编程范式开发算子基于算子模板库高效开发矩阵乘及相关融合算子

CATLASS 算子开发

基于算子模板库高效开发矩阵乘及相关融合算子

Triton 算子开发

面向Tile粒度的Python化编程,自动完成性能优化

获取资源

下载CANN软件包
使用Ascend C进行算子开发,需要安装CANN Toolkit开发套件包,完成环境准备

上手开发

HelloWorld 快速上手
快速上手Ascend C,完成矢量算子核函数和基础调用能力
开发流程 全面掌握
全面掌握Ascend C,完成Tiling开发,并进行单算子、网络中算子调用
高性能编程 深度驾驭
深度驾驭Ascend C,开发高性能的复杂算子,充分发挥硬件能力

拓展资源

融合算子加速库
提供大模型结构泛化的高性能融合算子
MindStudio Insight 可视化工具
面向昇腾AI开发者的可视化调优工具,辅助快速定位模型和算子的性能问题
代码样例
访问CANN及MindStudio样例仓
昇腾学习
打造一站式学习平台,助力开发者领跑AI开发之路
博客
技术沉淀,经验分享,记录思考与成长