Skip to content
版 本

asc_half2int16

产 品 支 持 情 况

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

功 能 说 明

将half类 型 数 据 转 换 为int16_t,并 支 持 多 种 舍 入 模 式。

  • RINT舍 入 模 式:四 舍 六 入 五 成 双 舍 入
  • ROUND舍 入 模 式:四 舍 五 入 舍 入
  • FLOOR舍 入 模 式:向 负 无 穷 舍 入
  • CEIL舍 入 模 式:向 正 无 穷 舍 入
  • TRUNC舍 入 模 式:向 零 舍 入

函 数 原 型

  • 前n个 数 据 计 算

    C++
    // RINT舍 入 模 式
    __aicore__ inline void asc_half2int16_rn(__ubuf__ int16_t* dst, __ubuf__ half* src, uint32_t count)
    // ROUND舍 入 模 式
    __aicore__ inline void asc_half2int16_rna(__ubuf__ int16_t* dst, __ubuf__ half* src, uint32_t count)
    //FLOOR舍 入 模 式
    __aicore__ inline void asc_half2int16_rd(__ubuf__ int16_t* dst, __ubuf__ half* src, uint32_t count)
    // CEIL舍 入 模 式
    __aicore__ inline void asc_half2int16_ru(__ubuf__ int16_t* dst, __ubuf__ half* src, uint32_t count)
    // TRUNC舍 入 模 式
    __aicore__ inline void asc_half2int16_rz(__ubuf__ int16_t* dst, __ubuf__ half* src, uint32_t count)
    
  • 高 维 切 分 计 算

    C++
    // RINT舍 入 模 式
    __aicore__ inline void asc_half2int16_rn(__ubuf__ int16_t* dst, __ubuf__ half* src, uint8_t repeat, uint16_t dst_block_stride, uint16_t src_block_stride, uint16_t dst_repeat_stride, uint16_t src_repeat_stride)
    // ROUND舍 入 模 式
    __aicore__ inline void asc_half2int16_rna(__ubuf__ int16_t* dst, __ubuf__ half* src, uint8_t repeat, uint16_t dst_block_stride, uint16_t src_block_stride, uint16_t dst_repeat_stride, uint16_t src_repeat_stride)
    // FLOOR舍 入 模 式
    __aicore__ inline void asc_half2int16_rd(__ubuf__ int16_t* dst, __ubuf__ half* src, uint8_t repeat, uint16_t dst_block_stride, uint16_t src_block_stride, uint16_t dst_repeat_stride, uint16_t src_repeat_stride)
    // CEIL舍 入 模 式
    __aicore__ inline void asc_half2int16_ru(__ubuf__ int16_t* dst, __ubuf__ half* src, uint8_t repeat, uint16_t dst_block_stride, uint16_t src_block_stride, uint16_t dst_repeat_stride, uint16_t src_repeat_stride)
    // TRUNC舍 入 模 式
    __aicore__ inline void asc_half2int16_rz(__ubuf__ int16_t* dst, __ubuf__ half* src, uint8_t repeat, uint16_t dst_block_stride, uint16_t src_block_stride, uint16_t dst_repeat_stride, uint16_t src_repeat_stride)
    
  • 同 步 计 算

    C++
    // RINT舍 入 模 式
    __aicore__ inline void asc_half2int16_rn_sync(__ubuf__ int16_t* dst, __ubuf__ half* src, uint32_t count)
    // ROUND舍 入 模 式
    __aicore__ inline void asc_half2int16_rna_sync(__ubuf__ int16_t* dst, __ubuf__ half* src, uint32_t count)
    // FLOOR舍 入 模 式
    __aicore__ inline void asc_half2int16_rd_sync(__ubuf__ int16_t* dst, __ubuf__ half* src, uint32_t count)
    // CEIL舍 入 模 式
    __aicore__ inline void asc_half2int16_ru_sync(__ubuf__ int16_t* dst, __ubuf__ half* src, uint32_t count)
    // TRUNC舍 入 模 式
    __aicore__ inline void asc_half2int16_rz_sync(__ubuf__ int16_t* dst, __ubuf__ half* src, uint32_t count)
    

参 数 说 明

参 数 名输 入/输 出描 述
dst输 出目 的 操 作 数(矢 量)的 起 始 地 址。
src输 入源 操 作 数(矢 量)的 起 始 地 址。
count输 入参 与 计 算 的 元 素 个 数。
repeat输 入迭 代 次 数。
dst_block_stride输 入目 的 操 作 数 单 次 迭 代 内 不 同DataBlock间 地 址 步 长。
src_block_stride输 入源 操 作 数 单 次 迭 代 内 不 同DataBlock间 地 址 步 长。
dst_repeat_stride输 入目 的 操 作 数 相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。
src_repeat_stride输 入源 操 作 数 相 邻 迭 代 间 相 同DataBlock的 地 址 步 长。

返 回 值 说 明

流 水 类 型

PIPE_V

约 束 说 明

调 用 示 例

C++
constexpr uint32_t total_length = 256;
__ubuf__ half src[total_length];
__ubuf__ int16_t dst[total_length];
asc_half2int16_rn(dst, src, total_length);

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