Skip to content
版 本

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);

免 责 声 明:本 站 内 容 由 asc-devkit 仓 master 分 支 自 动 编 译 生 成,属 于 持 续 开 发 版 本,可 能 存 在 缺 陷,仅 供 预 览 与 参 考。如 需 稳 定 及 商 用 资 料,请 查 阅 官 方 昇 腾 社 区