Skip to content
版 本

asc_bitsort

产 品 支 持 情 况

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

功 能 说 明

位 图 排 序 函 数,一 次 迭 代 可 以 完 成32个 元 素 的 排 序,数 据 按 如 下 描 述 结 构 进 行 保 存:

Score和Index分 别 存 储 在src0和src1中,按Score进 行 排 序(Score大 的 元 素 排 前 面),排 序 后 的Score与 其 对 应 的Index一 起 以(Score,Index)的 结 构 存 储 在dst中。

函 数 原 型

  • 连 续 计 算

    C++
    __aicore__ inline void asc_bitsort(__ubuf__ half* dst, __ubuf__ half* src0, __ubuf__ uint32_t* src1, int32_t repeat)
    __aicore__ inline void asc_bitsort(__ubuf__ float* dst, __ubuf__ float* src0, __ubuf__ uint32_t* src1, int32_t repeat)
    
  • 同 步 计 算

    C++
    __aicore__ inline void asc_bitsort_sync(__ubuf__ half* dst, __ubuf__ half* src0, __ubuf__ uint32_t* src1, int32_t repeat)
    __aicore__ inline void asc_bitsort_sync(__ubuf__ float* dst, __ubuf__ float* src0, __ubuf__ uint32_t* src1, int32_t repeat)
    

参 数 说 明

参 数 名输 入/输 出描 述
dst输 出目 的 操 作 数(矢 量)的 起 始 地 址。
src0输 入源 操 作 数(矢 量)的 起 始 地 址。
src1输 入源 操 作 数(矢 量)的 起 始 地 址。
repeat输 入重 复 迭 代 次 数,每 次 迭 代 完 成32个 元 素 的 排 序,下 次 迭 代src0和src1各 跳 过32个 元 素,dst跳 过256 Byte空 间。取 值 范 围:[0, 255]。

返 回 值 说 明

流 水 类 型

PIPE_V

约 束 说 明

调 用 示 例

C++
__ubuf__ half dst[256];
__ubuf__ half src0[256];
__ubuf__ uint32_t src1[256];
asc_bitsort(dst, src0, src1, 8);

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