对Tile执行原位缩放累加(AXPY,):将 [object Object] 按标量 [object Object] 缩放后累加到 [object Object] 上。
[object Object] 既是累加输入()也是输出,调用前必须已初始化;[object Object]()只读;[object Object]()为标量。
对于有效区域中的每个元素 [object Object]:
[object Object]:读-修改-写(RMW)。读入旧值作为累加基 ,写回 。[object Object]:只读,逐元素参与运算()。[object Object]:标量缩放系数(),类型为[object Object]。
[object Object]
声明于 [object Object]。
[object Object]
[object Object]
对于有效形状 :
[object Object]
[object Object]
TAXPY在向量流水线([object Object])上执行,使用 [object Object]()向量内建:
- 同类型(
[object Object]与[object Object]同dtype):逐repeat加载[object Object]与[object Object],执行[object Object]后写回[object Object];尾部不足一个repeat的列由谓词掩码屏蔽。 - 差异类型(
[object Object]=[object Object],[object Object]=[object Object]):[object Object]的half数据拓宽为FP32后参与累加(Ascend 950PR/Ascend 950DT上经[object Object]解包并[object Object]转换;Atlas A2/A3 训练系列产品/Atlas A2/A3 推理系列产品由[object Object]原生按4-block src / 8-block dst处理)。 - Atlas A2/A3 训练系列产品/Atlas A2/A3 推理系列产品上按repeat-stride是否溢出、以及列数与行数的关系,在count模式与norm模式间选择,以覆盖任意有效形状。
[object Object]
完整ST示例见 [object Object](A5)、[object Object](A2/A3)、[object Object](Kirin9030)及 [object Object](CPU参考实现)。