asc_reduce

产 品 支 持 情 况

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

功 能 说 明

以 内 置 固 定 模 式 对 应 的 二 进 制 或 者 用 户 自 定 义 输 入 的 数 值 对 应 的gather mask（数 据 收 集 的 掩 码），从 源 操 作 数 中 选 取 元 素 写 入 目 的 操 作 数 中。

函 数 原 型

• 前n个 数 据 计 算

  C++

  __aicore__ inline void asc_reduce(__ubuf__ uint16_t* dst, __ubuf__ uint16_t* src0, __ubuf__ uint16_t* src1, uint32_t count)
  __aicore__ inline void asc_reduce(__ubuf__ uint32_t* dst, __ubuf__ uint32_t* src0, __ubuf__ uint32_t* src1, uint32_t count)
  

• 高 维 切 分 计 算

  C++

  __aicore__ inline void asc_reduce(__ubuf__ uint16_t* dst, __ubuf__ uint16_t* src0, __ubuf__ uint16_t* src1,
    uint16_t repeat, uint8_t src0_block_stride, uint8_t pattern_mode, uint16_t src0_repeat_stride, uint8_t src1_repeat_stride)
  __aicore__ inline void asc_reduce(__ubuf__ uint32_t* dst, __ubuf__ uint32_t* src0, __ubuf__ uint32_t* src1,
    uint16_t repeat, uint8_t src0_block_stride, uint8_t pattern_mode, uint16_t src0_repeat_stride, uint8_t src1_repeat_stride)
  

• 同 步 计 算

  C++

  __aicore__ inline void asc_reduce_sync(__ubuf__ uint16_t* dst, __ubuf__ uint16_t* src0, __ubuf__ uint16_t* src1, uint32_t count)
  __aicore__ inline void asc_reduce_sync(__ubuf__ uint32_t* dst, __ubuf__ uint32_t* src0, __ubuf__ uint32_t* src1, uint32_t count)
  

参 数 说 明

表1参 数 说 明

参 数 名	输 入/输 出	描 述
dst	输 出	目 的 操 作 数（矢 量）的 起 始 地 址。
src0	输 入	源 操 作 数（矢 量）的 起 始 地 址。
src1	输 入	在 用 户 自 定 义 模 式 下，由 用 户 自 定 义 的 数 据 收 集 的 掩 码。
count	输 入	参 与 计 算 的 元 素 个 数。
repeat	输 入	迭 代 次 数。
src0_block_stride	输 入	源 操 作 数0单 次 迭 代 内 不 同DataBlock间 地 址 步 长。
pattern_mode	输 入	gather mask（数 据 收 集 的 掩 码），分 为 内 置 固 定 模 式 和 用 户 自 定 义 模 式 两 种，根 据 内 置 固 定 模 式 对 应 的 二 进 制 或 者 用 户 自 定 义 输 入 的 数 值 对 应 的 二 进 制 从 源 操 作 数 中 选 取 元 素 写 入 目 的 操 作 数 中。 •内 置 固 定 模 式：pattern_mode数 据 类 型 为uint8_t，取 值 范 围 为[1, 7]，所 有 迭 代 使 用 相 同 的gather mask。不 支 持 配 置src1。   • 1：01010101...0101 # 每 个repeat取 偶 数 索 引 元 素   • 2：10101010...1010 # 每 个repeat取 奇 数 索 引 元 素   • 3：00010001...0001 # 每 个repeat内 每 四 个 元 素 取 第 一 个 元 素   • 4：00100010...0010 # 每 个repeat每 四 个 元 素 取 第 二 个 元 素   • 5：01000100...0100 # 每 个repeat内 每 四 个 元 素 取 第 三 个 元 素   • 6：10001000...1000 # 每 个repeat内 每 四 个 元 素 区 第 四 个 元 素   • 7：11111111...1111 # 每 个repeat内 取 全 部 元 素 •用 户 自 定 义 模 式：pattern_mode取 值 为0时 为 用 户 自 定 义 模 式，用 户 通 过src1参 数 输 入 自 定 义 的gather mask，迭 代 间 隔 由src1_repeat_stride决 定。
src0_repeat_stride	输 入	源 操 作 数0相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。
src1_repeat_stride	输 入	源 操 作 数1相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。

返 回 值 说 明

无

流 水 类 型

PIPE_V

约 束 说 明

• 接 口 不 支 持 通 过asc_set_vector_mask配 置 掩 码。
• dst、src0、src1的 起 始 地 址 需 要32字 节 对 齐。
• 操 作 数 地 址 重 叠 约 束 请 参 考通 用 地 址 重 叠 约 束。

调 用 示 例

C++

//total_length指 参 与 计 算 的 数 据 总 长 度
constexpr uint64_t total_length = 128;
__ubuf__ uint16_t src0[total_length];
__ubuf__ uint16_t src1[total_length/16];
__ubuf__ uint16_t dst[total_length];
asc_reduce(dst, src0, src1, total_length);