体验昇腾 Ascend C 编程语言极简易用的算子开发
本文分享自华为云社区《CANN黑科技解密|昇腾Ascend C编程语言 — 极简易用的算子开发体验》,作者:昇腾 CANN 。
AI 应用的大脑是神经网络,而构成神经网络的基石是一个个算子。为了让开发者的网络在昇腾硬件上高效运行,昇腾异构计算架构 CANN(Compute Architecture for Neural Networks)提供了丰富的高性能算子库,包括神经网络库、线性代数计算库等,高性能算子数量达到 1400+。有了高性能算子库的支撑,主流神经网络可轻松在昇腾硬件上高效运行。
但随着人工智能的爆发式增长,算法更新层出不穷,固定的高性能算子库可能无法完全满足开发者的多样化创新需求,为了让开发者提出的创新算法能够在硬件上运行起来,自定义算子开发的能力必不可少。
但算子开发是个复杂的工程,需要考虑众多因素,包括语言学习成本、功能逻辑的实现、硬件指令的适配、以及算子运行精度与性能的达标等等。往往一个经验丰富的算法专家开发一个高性能的算子都要耗时数周甚至更长的时间。
算子开发总体流程
为提升算子开发效率,降低算子开发成本,昇腾推出了面向算子开发场景的昇腾 Ascend C 编程语言。昇腾 Ascend C 编程语言原生支持 C 和 C++标准规范,最大化匹配用户开发习惯;通过多层接口抽象,屏蔽了底层硬件差异;通过自动并行计算等关键技术,在保证性能的同时大大降低算子开发门槛。另外,昇腾 Ascend C 提供了孪生调试功能,大大缩短了算子调测时间。
Ascend C 编程语言在异构计算架构 CANN 中的位置
多层级高性能类库接口
工欲善其事,必先利其器。为简化开发逻辑,昇腾 Ascend C 支持结构化核函数编程,提供了面向不同场景的高性能类库接口。开发者仅需通过类库接口的组装调用,即可轻松实现高性能算子。
Ascend C 提供的类库接口
AI 应用领域广泛,开发者的背景与需求也各不相同。为满足不同层级开发者的诉求,昇腾 Ascend C 针对计算接口和数据搬移接口,进行了分层分级,让开发者可以根据自身需求选择合适的接口。
针对计算类接口,当前 Ascend C 支持三个层级,其中级数越低,自由度越高,更易于表达复杂场景所需功能;级数越高,接口的封装度越高,更易于表达常用语义,使用起来也更简单。
计算接口分层分级与示例
针对搬移类接口,Ascend C 将不同类型物理内存间的数据搬移抽象为一个统一的数据搬运接口,通过参数控制不同的搬运级别,从而满足不同数据搬运场景的需求。
数据搬运接口分层分级与示例
另外,多层级的类库接口封装,可以更好地屏蔽不同型号硬件间的差异,轻松实现算子代码对不同硬件的兼容。
自动并行计算
多层级的类库接口可以让开发者轻松实现算子的算法逻辑,达成预期功能。但一个好的算子,计算效率也是必须考虑的重要指标。众所周知,将任务并行处理是提高计算效率的关键手段,但 AI 处理器的内存层次结构比较复杂、数据通路多,数据之间的依赖关系复杂,这种场景下,并行计算之间的流水如何排布,各任务间的数据同步如何实现,往往是比较困难的。
为了方便开发者实现高效的并行计算,昇腾 Ascend C 采用 SPMD(Single-Program Multiple-Data)并行模式,开发者仅需关注一个计算核心上的算子程序实现,程序调用时,可自动启动 N 个运行实例(我们称之为 Block),每个实例都可部署到不同的计算核心上执行。由此,大大简化了开发者在多个计算核心上的并行编程逻辑。
单程序多数据 SPMD 并行计算
在算子逻辑实现上,Ascend C 基于流水线并行的编程范式,将算子核心逻辑划分为“搬入、计算、搬出”,开发者只需聚焦实现“搬入、计算、搬出”内容,程序运行时,系统会自动将核内数据进行分片,每一片数据都专注完成单一功能,实现计算性能最大化。
核内多片数据流水线并行调度
孪生调试
昇腾 Ascend C 提供的多层级类库接口以及自动并行计算功能,给开发者提供了轻松高效的编码体验。但在上一代算子开发的整个流程中,代码编写的时间往往仅占不到 30%,剩下 70%多的时间都在进行功能与性能调试,好的调试能力对提升端到端开发效率的重要性不言而喻。
开发时间占比示例
那为什么算子调试如此耗时呢?究其主要原因,一方面是由于 NPU 环境下本身调试困难;另一方面是因为编程过程隐藏了并行细节,导致同步死锁、地址越界、数据溢出等问题难定位。
为提升算子调试效率,Ascend C 提供了孪生调试能力,开发者既可以在 CPU 域进行调试调优,又可以在 NPU 侧进行调优验证,通过 CPU 域与 NPU 域相结合的方式,降低调试难度,提升调试效率。
在 CPU 域,开发者可通过业界标准 C++工具 GCC 编译器进行编译,并通过 GDB 通用调试工具进行单步调试,精准验证程序执行流程是否符合预期。另外,昇腾 Ascend C 还提供了主动 Bug 分析工具,方便开发者快速进行问题定位;在 NPU 域,昇腾 Ascend C 提供了仿真调试能力与上板调试能力,开发者可以通过仿真调试的 Profiling 流水图、指令日志以及数据日志,精准进行性能调优,也可以通过上板调试进行真实行为的验证。孪生调试的能力,在提升算子调试效率的同时又可保证精度与性能的达标。
Ascend C 孪生调试
人工智能蓬勃发展,算法创新层出不穷。昇腾提供的 Ascend C 编程语言,通过易上手、高性能、易调测的优势,为开发者的创新算法更轻松高效地在昇腾平台运行奠定了基石,让基于昇腾平台的 AI 创新更加简单。
点击如下链接,快速体验 Ascend C 算子开发
版权声明: 本文为 InfoQ 作者【华为云开发者联盟】的原创文章。
原文链接:【http://xie.infoq.cn/article/d3adb28d6ff8acc756bb813d2】。文章转载请联系作者。
评论