asc_datablock_reduce_sum
产 品 支 持 情 况
| 产 品 | 是 否 支 持 |
|---|---|
| √ | |
| √ |
功 能 说 明
对 每 个DataBlock内 所 有 元 素 求 和。
源 操 作 数 相 加 采 用 二 叉 树 的 方 式，两 两 相 加。以128个half类 型 数 据 求 和 为 例，每 个DataBlock可 以 计 算16个half类 型 数 据，分 成8个DataBlock计 算；每 个DataBlock内，通 过 二 叉 树 的 方 式，两 两 相 加。
需 要 注 意 的 是，两 两 相 加 计 算 过 程 中，计 算 结 果 大 于65504时 结 果 保 存 为65504。例 如，源 操 作 数 为[60000,60000,-30000,100]，首 先60000+60000溢 出，结 果 为65504，然 后 计 算-30000+100=-29900，最 后 计 算65504-29900=35604。
函 数 原 型
前n个 数 据 计 算
C++__aicore__ inline void asc_datablock_reduce_sum(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count) __aicore__ inline void asc_datablock_reduce_sum(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)高 维 切 分 计 算
C++__aicore__ inline void asc_datablock_reduce_sum(__ubuf__ half* dst, __ubuf__ half* src, uint8_t repeat, uint16_t dst_repeat_stride, uint16_t src_block_stride, uint16_t src_repeat_stride) __aicore__ inline void asc_datablock_reduce_sum(__ubuf__ float* dst, __ubuf__ float* src, uint8_t repeat, uint16_t dst_repeat_stride, uint16_t src_block_stride, uint16_t src_repeat_stride)同 步 计 算
C++__aicore__ inline void asc_datablock_reduce_sum_sync(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count) __aicore__ inline void asc_datablock_reduce_sum_sync(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)
参 数 说 明
表1参 数 说 明
| 参 数 名 | 输 入/输 出 | 描 述 |
|---|---|---|
| dst | 输 出 | 目 的 操 作 数（矢 量）的 起 始 地 址。 |
| src | 输 入 | 源 操 作 数（矢 量）的 起 始 地 址。 |
| count | 输 入 | 参 与 计 算 的 元 素 个 数。 |
| repeat | 输 入 | 迭 代 次 数。 |
| dst_repeat_stride | 输 入 | 目 的 操 作 数 相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。 输 入 类 型 位 宽 为16bit时，单 位 为16Byte，输 入 类 型 位 宽 为32bit时，单 位 为32Byte。 |
| src_block_stride | 输 入 | 源 操 作 数 单 次 迭 代 内 不 同DataBlock间 地 址 步 长。 |
| src_repeat_stride | 输 入 | 源 操 作 数 相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。 |
返 回 值 说 明
无
流 水 类 型
PIPE_V
约 束 说 明
- dst、src的 起 始 地 址 需 要32字 节 对 齐。
- 操 作 数 地 址 重 叠 约 束 请 参 考通 用 地 址 重 叠 约 束。
调 用 示 例
C++
constexpr uint32_t src_length = 256;
constexpr uint32_t dst_length = 16;
__ubuf__ half src[src_length];
__ubuf__ half dst[dst_length];
// 每 次repeat256B，2次repeat，无 间 隔
asc_datablock_reduce_sum(dst, src, 2, 1, 1, 8);
结 果 示 例：
Text
输 入 数 据src：[1 1 1 ... 1]
输 出 数 据dst：[16 16 16 ... 16]