编程模型概述

AI Core是AI处理器的计算核心,AI处理器通过多个AI Core实现并行计算。与传统CPU相比,AI处理器因其内部拥有更多的计算单元和相应的向量计算指令,更适合模型训练和推理场景,这使得单个硬件指令能完成多组数据的计算。AI处理器提供了以下两种编程模型:

AI CPU是位于Device侧的处理器,具备与AI Core相同的内存访问能力,能够直接访问Device侧的内存资源;同时,它也可以像Host侧的CPU一样进行数据计算。

表1 编程模型分类

编程模型

计算空间

特点

SIMD编程

AI Core

适合矩阵计算、连续计算的矢量算子及融合算子场景,提供SIMD与SIMT混合的高级编程方式。

SIMT编程

AI Core

适用于离散访问场景、复杂分支控制场景。

AI CPU编程

AI CPU

作为AI Core计算的补充。