Skip to content
版 本

asc_repeat_reduce_min

产 品 支 持 情 况

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

功 能 说 明

对 每 个Repeat内 所 有 元 素 求 最 小 值,以 及Repeat内 的 首 个 最 小 值 元 素 在 当 前Repeat中 的 索 引index。一 次Repeat计 算256字 节。

index_value接 口:表 示index位 于 低 半 部,返 回 结 果 存 储 顺 序 为[index, value]。输 入 数 据 类 型 为half时,index数 据 类 型 为uint16;输 入 数 据 类 型 为float时,index数 据 类 型 为uint32。 value_index接 口:表 示value位 于 低 半 部,返 回 结 果 存 储 顺 序 为[value, index]。输 入 数 据 类 型 为half时,index数 据 类 型 为uint16;输 入 数 据 类 型 为float时,index数 据 类 型 为uint32。 only_value接 口:表 示 只 返 回 最 值,返 回 结 果 存 储 顺 序 为[value]。 only_index接 口:表 示 只 返 回 最 值 索 引,返 回 结 果 存 储 顺 序 为[index]。index数 据 类 型 固 定 为uint32

执 行 此 接 口 后,可 以 继 续 执 行asc_get_reduce_min_cnt获 取 全 部repeat内 的 最 小 值,以 及 第 一 个 最 小 值 时 的 索 引。

函 数 原 型

  • 前n个 数 据 计 算

    C++
    __aicore__ inline void asc_repeat_reduce_min_index_value(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_index_value(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_value_index(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_value_index(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_only_value(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_only_value(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_only_index(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_only_index(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)
    
  • 高 维 切 分 计 算

    C++
    __aicore__ inline void asc_repeat_reduce_min_index_value(__ubuf__ half* dst, __ubuf__ half* src, uint8_t repeat, uint16_t dst_repeat_stride, uint16_t src_block_stride, uint16_t src_repeat_stride)
    __aicore__ inline void asc_repeat_reduce_min_index_value(__ubuf__ float* dst, __ubuf__ float* src, uint8_t repeat, uint16_t dst_repeat_stride, uint16_t src_block_stride, uint16_t src_repeat_stride)
    __aicore__ inline void asc_repeat_reduce_min_value_index(__ubuf__ half* dst, __ubuf__ half* src, uint8_t repeat, uint16_t dst_repeat_stride, uint16_t src_block_stride, uint16_t src_repeat_stride)
    __aicore__ inline void asc_repeat_reduce_min_value_index(__ubuf__ float* dst, __ubuf__ float* src, uint8_t repeat, uint16_t dst_repeat_stride, uint16_t src_block_stride, uint16_t src_repeat_stride)
    __aicore__ inline void asc_repeat_reduce_min_only_value(__ubuf__ half* dst, __ubuf__ half* src, uint8_t repeat, uint16_t dst_repeat_stride, uint16_t src_block_stride, uint16_t src_repeat_stride)
    __aicore__ inline void asc_repeat_reduce_min_only_value(__ubuf__ float* dst, __ubuf__ float* src, uint8_t repeat, uint16_t dst_repeat_stride, uint16_t src_block_stride, uint16_t src_repeat_stride)
    __aicore__ inline void asc_repeat_reduce_min_only_index(__ubuf__ half* dst, __ubuf__ half* src, uint8_t repeat, uint16_t dst_repeat_stride, uint16_t src_block_stride, uint16_t src_repeat_stride)
    __aicore__ inline void asc_repeat_reduce_min_only_index(__ubuf__ float* dst, __ubuf__ float* src, uint8_t repeat, uint16_t dst_repeat_stride, uint16_t src_block_stride, uint16_t src_repeat_stride)
    
  • 同 步 计 算

    C++
    __aicore__ inline void asc_repeat_reduce_min_index_value_sync(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_index_value_sync(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_value_index_sync(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_value_index_sync(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_only_value_sync(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_only_value_sync(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_only_index_sync(__ubuf__ half* dst, __ubuf__ half* src, uint32_t count)
    __aicore__ inline void asc_repeat_reduce_min_only_index_sync(__ubuf__ float* dst, __ubuf__ float* src, uint32_t count)
    

参 数 说 明

表1参 数 说 明

参 数 名输 入/输 出描 述
dst输 出目 的 操 作 数(矢 量)的 起 始 地 址。
src输 入源 操 作 数(矢 量)的 起 始 地 址。
count输 入参 与 计 算 的 元 素 个 数。
repeat输 入迭 代 次 数。
dst_repeat_stride输 入目 的 操 作 数 相 邻 迭 代 间 的 地 址 步 长,单 位 为 一 个repeat计 算 后 的 长 度,具 体 如 下:
返 回 最 值 和 索 引 时,若 输 出 数 据 类 型 为T,则 单 位 为sizeof(T)*2。
仅 返 回 最 值 且 输 入 类 型 为T时,单 位 为sizeof(T)。
仅 返 回 索 引 时,单 位 为4字 节(sizeof(uint32))
src_block_stride输 入源 操 作 数 单 次 迭 代 内 不 同DataBlock间 地 址 步 长。
src_repeat_stride输 入源 操 作 数 相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。

返 回 值 说 明

流 水 类 型

PIPE_V

约 束 说 明

  • 输 出 中 的 索 引index的 数 据 是 按 照dst的 数 据 类 型 进 行 存 储,比 如float输 出 的index数 据 应 该 为uint32,则 读 取 时 需 要 将float重 新 解 释 为uint32。
  • 操 作 数 地 址 重 叠 约 束 请 参 考通 用 地 址 重 叠 约 束
  • dst的 起 始 地 址,输 入 类 型 为half时 需 要4字 节 对 齐,输 入 类 型 为float时 需 要8字 节 对 齐。
  • src的 起 始 地 址 需 要32字 节 对 齐。

调 用 示 例

C++
// 示 例 计 算256个half的 最 小 值 和 索 引,分2次 迭 代 完 成,计 算 结 果 为2组value_index
constexpr uint32_t total_length = 256;
__ubuf__ half src[total_length];
constexpr uint32_t dst_length = 4;
__ubuf__ half dst[dst_length];

uint8_t repeat = 2;
// 单 位 为sizeof(half)*2 = 4字 节,连 续 存 放
uint16_t dst_repeat_stride = 1;
uint16_t src_block_stride = 1;
uint16_t src_repeat_stride = 8;
asc_repeat_reduce_min_value_index(dst, src, repeat, dst_repeat_stride, src_block_stride, src_repeat_stride);

结 果 示 例:

Text
输 入 数 据src:[256, 255, 254, ..., 129, 128, ..., 2, 1]
输 出 数 据dst:[129, 7.6e-6, 1, 7.6e-6]
输 出 说 明:half类 型7.6e-6的 二 进 制 是0x007F,重 新 解 释 为uint32后 即 为 最 小 值 索 引127

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