Skip to content
版 本

asc_ge

产 品 支 持 情 况

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

功 能 说 明

Ge(greater than or equal to),逐 元 素 比 较src0 >= src1是 否 成 立,成 立 则 输 出 结 果 为1,否 则 输 出 结 果 为0,每 个 元 素 的 比 较 结 果 占 一 个bit。计 算 公 式 如 下:

$$ dst = src0 >= src1 $$

函 数 原 型

  • 高 维 切 分 计 算

    C++
    // 比 较 结 果 存 入 寄 存 器
    __aicore__ inline void asc_ge(__ubuf__ half* src0, __ubuf__ half* src1, uint8_t repeat, uint8_t dst_block_stride,
        uint8_t src0_block_stride, uint8_t src1_block_stride, uint8_t dst_repeat_stride, uint8_t src0_repeat_stride,
        uint8_t src1_repeat_stride);
    __aicore__ inline void asc_ge(__ubuf__ float* src0, __ubuf__ float* src1, uint8_t repeat, uint8_t dst_block_stride,
        uint8_t src0_block_stride, uint8_t src1_block_stride, uint8_t dst_repeat_stride, uint8_t src0_repeat_stride,
        uint8_t src1_repeat_stride);
    // 比 较 结 果 存 入dst
    __aicore__ inline void asc_ge(__ubuf__ uint8_t* dst, __ubuf__ half* src0, __ubuf__ half* src1, uint8_t repeat, 
        uint8_t dst_block_stride, uint8_t src0_block_stride, uint8_t src1_block_stride, uint8_t dst_repeat_stride, 
        uint8_t src0_repeat_stride, uint8_t src1_repeat_stride);
    __aicore__ inline void asc_ge(__ubuf__ uint8_t* dst, __ubuf__ float* src0, __ubuf__ float* src1, uint8_t repeat, 
        uint8_t dst_block_stride, uint8_t src0_block_stride, uint8_t src1_block_stride, uint8_t dst_repeat_stride, 
        uint8_t src0_repeat_stride, uint8_t src1_repeat_stride);
    
  • 同 步 计 算

    C++
    // 比 较 结 果 存 入 寄 存 器
    __aicore__ inline void asc_ge_sync(__ubuf__ half* src0, __ubuf__ half* src1, uint8_t repeat, uint8_t dst_block_stride, 
        uint8_t src0_block_stride, uint8_t src1_block_stride, uint8_t dst_repeat_stride, uint8_t src0_repeat_stride, 
        uint8_t src1_repeat_stride);
    __aicore__ inline void asc_ge_sync(__ubuf__ float* src0, __ubuf__ float* src1, uint8_t repeat, uint8_t dst_block_stride, 
        uint8_t src0_block_stride, uint8_t src1_block_stride, uint8_t dst_repeat_stride, uint8_t src0_repeat_stride, 
        uint8_t src1_repeat_stride);
    // 比 较 结 果 存 入dst
    __aicore__ inline void asc_ge_sync(__ubuf__ uint8_t* dst, __ubuf__ half* src0, __ubuf__ half* src1, uint8_t repeat, 
        uint8_t dst_block_stride, uint8_t src0_block_stride, uint8_t src1_block_stride, uint8_t dst_repeat_stride, 
        uint8_t src0_repeat_stride, uint8_t src1_repeat_stride);
    __aicore__ inline void asc_ge_sync(__ubuf__ uint8_t* dst, __ubuf__ float* src0, __ubuf__ float* src1, uint8_t repeat, 
        uint8_t dst_block_stride, uint8_t src0_block_stride, uint8_t src1_block_stride, uint8_t dst_repeat_stride, 
        uint8_t src0_repeat_stride, uint8_t src1_repeat_stride);
    

参 数 说 明

参 数 名输 入/输 出描 述
dst输 出目 的 操 作 数(矢 量)的 起 始 地 址。
src0、src1输 入源 操 作 数(矢 量)的 起 始 地 址。
repeat输 入迭 代 次 数。
dst_block_stride输 入目 的 操 作 数 单 次 迭 代 内 不 同DataBlock间 地 址 步 长。
src0_block_stride输 入源 操 作 数0单 次 迭 代 内 不 同DataBlock间 地 址 步 长。
src1_block_stride输 入源 操 作 数1单 次 迭 代 内 不 同DataBlock间 地 址 步 长。
dst_repeat_stride输 入目 的 操 作 数 相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。
src0_repeat_stride输 入源 操 作 数0相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。
src1_repeat_stride输 入源 操 作 数1相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。

返 回 值 说 明

流 水 类 型

PIPE_V

约 束 说 明

  • 操 作 数 地 址 重 叠 约 束 请 参 考通 用 地 址 重 叠 约 束
  • dst、src0、src1的 起 始 地 址 需 要32字 节 对 齐。
  • dst按 照 小 端 顺 序 排 序 成 二 进 制 结 果,对 应src中 相 应 位 置 的 数 据 比 较 结 果。
  • 当 结 果 输 出 到 目 的 地 址 中 时,mask参 数 不 生 效。
  • 使 用 寄 存 器 形 式 的 接 口 时,需 要 配 合asc_get_cmp_mask获 取 比 较 结 果。
  • 当 结 果 输 出 到 寄 存 器 中 时,固 定 处 理256B的 数 据,repeat字 段 不 生 效。
  • 当 结 果 输 出 到 寄 存 器 中 时,若mask某bit位 为0,对 应 输 出 寄 存 器 的bit位 的 值 不 会 改 变 时,最 终 结 果 可 能 会 有 上 一 次 计 算 的 残 留。如 果 要 初 始 化 寄 存 器,可 以 调 用asc_set_cmp_mask接 口。

调 用 示 例

C++
// 输 入 固 定 为128个 元 素
constexpr uint32_t total_length = 128;
__ubuf__ uint8_t dst[total_length / 8];
__ubuf__ half src0[total_length];
__ubuf__ half src1[total_length];
uint8_t repeat = 1;
uint8_t dst_block_stride = 1;
uint8_t src0_block_stride = 1;
uint8_t src1_block_stride = 1;
uint8_t dst_repeat_stride = 8;
uint8_t src0_repeat_stride = 8;
uint8_t src1_repeat_stride = 8;
// 结 果 存 入 寄 存 器,通 过 寄 存 器 获 取 比 较 结 果
asc_ge(src0, src1, repeat, dst_block_stride, src0_block_stride, src1_block_stride, dst_repeat_stride, src0_repeat_stride, src1_repeat_stride);
// 结 果 输 出 到 目 标 地 址 中
asc_ge(dst, src0, src1, repeat, dst_block_stride, src0_block_stride, src1_block_stride, dst_repeat_stride, src0_repeat_stride, src1_repeat_stride);

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