asc_ne_scalar

产 品 支 持 情 况

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

功 能 说 明

src中 的 每 个 元 素 逐 个 与 标 量value比 较 大 小，如 果 某 个 位 置 上 的 元 素 不 等 于value，则 输 出 结 果dst对 应 比 特 位 为1，否 则 为0。

函 数 原 型

• 高 维 切 分 计 算

  C++

  __aicore__ inline void asc_ne_scalar(__ubuf__ uint8_t *dst, __ubuf__ half *src, half value, uint8_t repeat, uint16_t dst_block_stride, uint16_t src_block_stride, uint16_t dst_repeat_stride, uint16_t src_repeat_stride)
  __aicore__ inline void asc_ne_scalar(__ubuf__ uint8_t *dst, __ubuf__ float *src, float value, uint8_t repeat, uint16_t dst_block_stride, uint16_t src_block_stride, uint16_t dst_repeat_stride, uint16_t src_repeat_stride)
  

• 同 步 计 算

  C++

  __aicore__ inline void asc_ne_scalar_sync(__ubuf__ uint8_t *dst, __ubuf__ half *src, half value, uint8_t repeat, uint16_t dst_block_stride, uint16_t src_block_stride, uint16_t dst_repeat_stride, uint16_t src_repeat_stride)
  __aicore__ inline void asc_ne_scalar_sync(__ubuf__ uint8_t *dst, __ubuf__ float *src, float value, uint8_t repeat, uint16_t dst_block_stride, uint16_t src_block_stride, uint16_t dst_repeat_stride, uint16_t src_repeat_stride)
  

参 数 说 明

参 数 名	输 入/输 出	描 述
dst	输 出	目 的 操 作 数（矢 量）的 起 始 地 址。
src	输 入	源 操 作 数（矢 量）的 起 始 地 址。
value	输 入	源 操 作 数（标 量）。
repeat	输 入	迭 代 次 数。
dst_block_stride	输 入	目 的 操 作 数 单 次 迭 代 内 不 同DataBlock间 地 址 步 长。
src_block_stride	输 入	源 操 作 数 单 次 迭 代 内 不 同DataBlock间 地 址 步 长。
dst_repeat_stride	输 入	目 的 操 作 数 相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。
src_repeat_stride	输 入	源 操 作 数 相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。

返 回 值 说 明

无

流 水 类 型

PIPE_V

约 束 说 明

• 操 作 数 地 址 重 叠 约 束 请 参 考通 用 地 址 重 叠 约 束。
• dst、src的 起 始 地 址 需 要32字 节 对 齐。
• dst按 照 小 端 顺 序 排 序 成 二 进 制 结 果，对 应src中 相 应 位 置 的 数 据 比 较 结 果。

调 用 示 例

C++

// 结 果 输 出 到 目 标 地 址 中，固 定 为128个 元 素
constexpr uint32_t total_length = 128;
__ubuf__ uint8_t dst[total_length / 8];
__ubuf__ half src[total_length];
half scalar = 20;
uint8_t repeat = 1;
uint8_t dst_block_stride = 1;
uint8_t src_block_stride = 1;
uint8_t dst_repeat_stride = 8;
uint8_t src_repeat_stride = 8;
…… // 数 据 搬 运 及 同 步 操 作
asc_ne_scalar(dst, src, scalar, repeat, dst_block_stride, src_block_stride, dst_repeat_stride, src_repeat_stride);
…… // 同 步 操 作