asc_gather
产 品 支 持 情 况
| 产 品 | 是 否 支 持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
功 能 说 明
根 据 索 引 位 置index将 源 操 作 数src按 元 素 收 集 到 目 的 操 作 数dst中。
函 数 原 型
收 集UB中 的 元 素
C++__simd_callee__ inline void asc_gather(vector_int16_t& dst, __ubuf__ int8_t* src, vector_uint16_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_uint16_t& dst, __ubuf__ uint8_t* src, vector_uint16_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_int16_t& dst, __ubuf__ int16_t* src, vector_uint16_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_uint16_t& dst, __ubuf__ uint16_t* src, vector_uint16_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_int32_t& dst, __ubuf__ int32_t* src, vector_uint32_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_uint32_t& dst, __ubuf__ uint32_t* src, vector_uint32_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_half& dst, __ubuf__ half* src, vector_uint16_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_float& dst, __ubuf__ float* src, vector_uint32_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_bfloat16_t& dst, __ubuf__ bfloat16_t* src, vector_uint16_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_fp8_e4m3fn_t& dst, __ubuf__ fp8_e4m3fn_t* src, vector_uint16_t index,vector_bool mask) __simd_callee__ inline void asc_gather(vector_fp8_e5m2_t& dst, __ubuf__ fp8_e5m2_t* src, vector_uint16_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_fp8_e8m0_t& dst, __ubuf__ fp8_e8m0_t* src, vector_uint16_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_hifloat8_t& dst, __ubuf__ hifloat8_t* src, vector_uint16_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_int16_t& dst, __ubuf__ int16_t* src, vector_uint32_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_uint16_t& dst, __ubuf__ uint16_t* src, vector_uint32_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_half& dst, __ubuf__ half* src, vector_uint32_t index, vector_bool mask) __simd_callee__ inline void asc_gather(vector_bfloat16_t& dst, __ubuf__ bfloat16_t* src, vector_uint32_t index, vector_bool mask)收 集 矢 量 数 据 寄 存 器 中 的 元 素
C++__simd_callee__ inline void asc_gather(vector_int8_t& dst, vector_int8_t src, vector_uint8_t index) __simd_callee__ inline void asc_gather(vector_uint8_t& dst, vector_uint8_t src, vector_uint8_t index) __simd_callee__ inline void asc_gather(vector_int16_t& dst, vector_int16_t src, vector_uint16_t index) __simd_callee__ inline void asc_gather(vector_uint16_t& dst, vector_uint16_t src, vector_uint16_t index) __simd_callee__ inline void asc_gather(vector_int32_t& dst, vector_int32_t src, vector_uint32_t index) __simd_callee__ inline void asc_gather(vector_uint32_t& dst, vector_uint32_t src, vector_uint32_t index) __simd_callee__ inline void asc_gather(vector_half& dst, vector_half src, vector_uint16_t index) __simd_callee__ inline void asc_gather(vector_bfloat16_t& dst, vector_bfloat16_t src, vector_uint16_t index) __simd_callee__ inline void asc_gather(vector_fp8_e4m3fn_t& dst, vector_fp8_e4m3fn_t src, vector_uint8_t index) __simd_callee__ inline void asc_gather(vector_fp8_e5m2_t& dst, vector_fp8_e5m2_t src, vector_uint8_t index) __simd_callee__ inline void asc_gather(vector_fp8_e8m0_t& dst, vector_fp8_e8m0_t src, vector_uint8_t index) __simd_callee__ inline void asc_gather(vector_hifloat8_t& dst, vector_hifloat8_t src, vector_uint8_t index)
参 数 说 明
收 集UB中 的 元 素
参 数 名 输 入/输 出 描 述 dst 输 出 目 的 操 作 数（矢 量 数 据 寄 存 器）。 src 输 入 源 操 作 数（矢 量）的 起 始 地 址。 index 输 入 数 据 索 引（矢 量 数 据 寄 存 器），dst中 每 个 元 素 在UB中 相 对 于src的 索 引 位 置。 mask 输 入 源 操 作 数 掩 码（掩 码 寄 存 器），用 于 指 示 在 计 算 过 程 中 哪 些 元 素 参 与 计 算。对 应 位 置 为1时 参 与 计 算，为0时 不 参 与 计 算。mask未 筛 选 的 元 素 在 输 出 中 置 零。 收 集 矢 量 数 据 寄 存 器 中 的 元 素
参 数 名 输 入/输 出 描 述 dst 输 出 目 的 操 作 数（矢 量 数 据 寄 存 器）。 src 输 入 源 操 作 数（矢 量 数 据 寄 存 器）。 index 输 入 数 据 索 引（矢 量 数 据 寄 存 器）。
矢 量 数 据 寄 存 器 和 掩 码 寄 存 器 的 详 细 说 明 请 参 见reg数 据 类 型 定 义.md。
返 回 值 说 明
无
流 水 类 型
PIPE_V
约 束 说 明
收 集UB中 的 元 素
当src的 数 据 类 型 为b8数 据 类 型 时，dst为b16数 据 类 型，这 种 情 况 下 目 的 操 作 数 的 低8位 与 源 操 作 数 相 同，高8位 自 动 补0。例 如src为int8_t数 据 类 型：
40 = 0b00101000 -> 0b0000000000101000,扩 充 至16位 后 等 于40；
-40 = 0b11011000 -> 0b0000000011011000，扩 充 至16位 后 等 于216。
收 集 矢 量 数 据 寄 存 器 中 的 元 素
src为 矢 量 数 据 寄 存 器 类 型，位 宽 是 固 定 的VL,存 储 的 元 素 个 数 固 定。如 果index中 索 引 值 超 出 当 前 矢 量 数 据 寄 存 器 中 能 存 储 的 最 大 元 素 个 数 时，按 照 如 下 方 式 处 理：设 定 当 前 矢 量 数 据 寄 存 器 所 能 存 储 的 最 大 数 据 元 素 个 数 为vl_length, index中 索 引 值 为i，索 引 值 更 新 为i % vl_length。
调 用 示 例
收 集UB中 的 元 素
C++vector_half dst; __ubuf__ half* src = (__ubuf__ half*)asc_get_phy_buf_addr(0); vector_uint16_t index; vector_bool mask = asc_create_mask_b16(PAT_ALL); asc_loadalign(index, index_addr); // index_addr是 外 部 输 入 的UB内 存 空 间 地 址。 asc_gather(dst, src, index, mask);收 集 矢 量 数 据 寄 存 器 中 的 元 素
C++vector_half dst; vector_half src; vector_uint16_t index; asc_loadalign(src, src_addr); // src_addr是 外 部 输 入 的UB内 存 空 间 地 址。 asc_loadalign(index, index_addr); // index_addr是 外 部 输 入 的UB内 存 空 间 地 址。 asc_gather(dst, src, index);