asc_datablock_reduce_max

产 品 支 持 情 况

产 品	是 否 支 持
Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品	√
Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品	√

功 能 说 明

执 行 数 据 块 内 的 求 最 大 值 规 约（Reduce Max）操 作。

函 数 原 型

• 前n个 数 据 连 续 计 算

  C++

  __aicore__ inline void asc_datablock_reduce_max(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count)
  __aicore__ inline void asc_datablock_reduce_max(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)
  

• 高 维 切 分 计 算

  C++

  __aicore__ inline void asc_datablock_reduce_max(__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_max(__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_max_sync(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count)
  __aicore__ inline void asc_datablock_reduce_max_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_max(dst, src, 2, 1, 1, 8);

结 果 示 例：

Text

输 入 数 据src：[1 2 3 ... 16 17 ... 32 ... 225... 256]
输 出 数 据dst：[16 32 ... 256]