asc_select
产 品 支 持 情 况
| 产 品 | 是 否 支 持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
功 能 说 明
根 据mask的 比 特 位 值,从 源 操 作 数src0,src1中 选 择 元 素,得 到 目 的 操 作 数dst。选 择 的 规 则 为:当mask的 比 特 位 为1时,从src0中 选 取 对 应 位 置 的 数;当mask的 比 特 位 为0时,从src1中 选 取 对 应 位 置 的 数。
计 算 公 式 如 下:
$$ dst_i = (mask_i == 1) ? src0_i : src1_i $$
函 数 原 型
C++
__simd_callee__ inline void asc_select(vector_bool& dst, vector_bool src0, vector_bool src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_uint8_t& dst, vector_uint8_t src0, vector_uint8_t src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_int8_t& dst, vector_int8_t src0, vector_int8_t src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_fp8_e4m3fn_t& dst, vector_fp8_e4m3fn_t src0, vector_fp8_e4m3fn_t src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_hifloat8_t& dst, vector_hifloat8_t src0, vector_hifloat8_t src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_fp8_e5m2_t& dst, vector_fp8_e5m2_t src0, vector_fp8_e5m2_t src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_uint16_t& dst, vector_uint16_t src0, vector_uint16_t src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_int16_t& dst, vector_int16_t src0, vector_int16_t src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_half& dst, vector_half src0, vector_half src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_bfloat16_t& dst, vector_bfloat16_t src0, vector_bfloat16_t src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_uint32_t& dst, vector_uint32_t src0, vector_uint32_t src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_int32_t& dst, vector_int32_t src0, vector_int32_t src1, vector_bool mask)
__simd_callee__ inline void asc_select(vector_float& dst, vector_float src0, vector_float src1, vector_bool mask)
参 数 说 明
| 参 数 名 | 输 入/输 出 | 描 述 |
|---|---|---|
| dst | 输 出 | 目 的 操 作 数(掩 码 寄 存 器 或 矢 量 数 据 寄 存 器)。 |
| src0 | 输 入 | 源 操 作 数0(掩 码 寄 存 器 或 矢 量 数 据 寄 存 器)。 |
| src1 | 输 入 | 源 操 作 数1(掩 码 寄 存 器 或 矢 量 数 据 寄 存 器)。 |
| mask | 输 入 | 源 操 作 数 掩 码(掩 码 寄 存 器)。指 定 选 择src0或src1为 有 效 数 据。mask的 比 特 位 为1时,选 取src0;mask的 比 特 位 为0时,选 取src1。 |
矢 量 数 据 寄 存 器 和 掩 码 寄 存 器 的 详 细 说 明 请 参 见reg数 据 类 型 定 义.md。
返 回 值 说 明
无
流 水 类 型
PIPE_V
约 束 说 明
无
调 用 示 例
C++
vector_half dst;
vector_half src0;
vector_half src1;
vector_bool mask = asc_create_mask_b16(PAT_ALL);
asc_loadalign(src0, src0_addr); // src0_addr是 外 部 输 入 的UB内 存 空 间 地 址。
asc_loadalign(src1, src1_addr); // src1_addr是 外 部 输 入 的UB内 存 空 间 地 址。
// mask为 源 操 作 数 掩 码
asc_select(dst, src0, src1, mask);