Skip to content
版本

asc_pair_reduce_sum

产品支持情况

  • Ascend 950PR/Ascend 950DT:支持
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:不支持
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:不支持
  • Atlas 200I/500 A2 推理产品:不支持
  • Atlas 推理系列产品AI Core:不支持
  • Atlas 推理系列产品Vector Core:不支持
  • Atlas 训练系列产品:不支持

功能说明

将传入的源操作数src中每两个相邻数值相加,并将相加的结果连续保存在目的操作数dst中的低半部分。以half数据类型为例,相加示意图如下:

asc_pair_reduce_sum示意图

函数原型

C++
__simd_callee__ inline void asc_pair_reduce_sum(vector_half& dst, vector_half src, vector_bool mask)
__simd_callee__ inline void asc_pair_reduce_sum(vector_float& dst, vector_float src, vector_bool mask)

参数说明

表1 参数说明

参数名输入/输出描述
dst输出目的操作数(矢量数据寄存器)。
src输入源操作数(矢量数据寄存器)。
mask输入源操作数掩码(掩码寄存器),用于指示在计算过程中哪些元素参与计算。对应位置为1时参与计算,为0时不参与计算,不参与计算的元素相加时被当作0处理。

矢量数据寄存器和掩码寄存器的详细说明请参见reg数据类型定义

返回值说明

约束说明

  • mask指定源操作数是否参与计算,不参与计算的元素,被当作0进行累加。例如:

    • 若连续两个元素a、b均不参与计算,目的操作数结果为0。
    • 若连续两个元素a、b仅元素a参与计算,目的操作数结果为a。
    • 若连续两个元素a、b仅元素b参与计算,目的操作数结果为b。
  • 求和后,dst中仅前一半元素为有效数据,后一半填充为0。搬出至UB时需要避免数据踩踏。

调用示例

C++
vector_half dst;
vector_half src;
vector_bool mask = asc_create_mask_b16(PAT_ALL);
asc_loadalign(src, src_addr); // src_addr是外部输入的UB内存空间地址。
asc_pair_reduce_sum(dst, src, mask);

免责声明:本站内容由 asc-devkit 仓 master 分支自动编译生成,属于持续开发版本,可能存在缺陷,仅供预览与参考。如需稳定及商用资料,请查阅官方 昇腾社区