Skip to content

asc_float2bfloat16(废弃)

产品支持情况

  • Ascend 950PR/Ascend 950DT:支持
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:不支持
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:不支持
  • Atlas 200I/500 A2 推理产品:不支持
  • Atlas 推理系列产品AI Core:不支持
  • Atlas 推理系列产品Vector Core:不支持
  • Atlas 训练系列产品:不支持

功能说明

头文件路径为:"c_api/reg_compute/reg_convert.h"

该接口通过接口后缀来控制目的操作数写入位置(仅包括源操作数、目的操作数、掩码寄存器三个参数)的原型已废弃,请使用asc_float2bfloat16的显式位置参数重载替代。

根据masksrc中的每个float类型数据转换成bfloat16_t类型,并支持多种舍入模式与饱和模式,结果写入dst。 由于源操作数与目的操作数类型位宽比为2:1,写入数据时需要将一个VL大小的数据分为两部分,根据不同接口选择数据写入索引为奇数的位置或偶数的位置。伪代码如下:

Python
def asc_float2bfloat16_rn(dst, src, mask):
    for i in range(64):
        if mask[i]:
            dst[2 * i] = bfloat16(src[i])   # float -> bfloat16,RINT舍入,写入偶数索引
        else:
            dst[2 * i] = 0
    for i in range(64):
        dst[2 * i + 1] = 0                  # 奇数索引置零

def asc_float2bfloat16_rn_v2(dst, src, mask):
    for i in range(64):
        dst[2 * i] = 0                      # 偶数索引置零
        if mask[i]:
            dst[2 * i + 1] = bfloat16(src[i])  # float -> bfloat16,RINT舍入,写入奇数索引
        else:
            dst[2 * i + 1] = 0

关于舍入模式的详细说明,请参见舍入模式

迁移说明

表1 新旧位置参数对应关系

旧原型位置后缀新原型位置参数
无位置后缀ASC_POSITION_EVEN
_v2ASC_POSITION_ODD

迁移时移除函数名中的位置后缀,并将对应位置常量作为第四个参数传入。新原型请参见asc_float2bfloat16

函数原型

模板原型(占位符形式)

C
__simd_callee__ inline void asc_float2bfloat16_<round_mode><sat_mode><position_mode>(vector_bfloat16_t& dst,
                                                                                     vector_float src,
                                                                                     vector_bool mask)

占位符说明

  • <round_mode>表示支持的舍入模式,支持rdFLOOR)、rnRINT)、rnaROUND)、ruCEIL)和rzTRUNC)。
  • <sat_mode>表示饱和/非饱和模式,为空时表示非饱和模式,取值为_sat时表示饱和模式。
  • <position_mode>表示dst写入的位置,为空时,结果写入dst的偶数索引位置;取值为_v2时,结果写入dst的奇数索引位置。

函数原型典型示例

C
// RINT舍入模式,非饱和模式,数据写入索引为偶数的位置
__simd_callee__ inline void asc_float2bfloat16_rn(vector_bfloat16_t& dst,
                                                  vector_float src,
                                                  vector_bool mask)

参数说明

表2 参数说明

参数名输入/输出描述
dst输出目的操作数(矢量数据寄存器)。
src输入源操作数(矢量数据寄存器)。
mask输入掩码寄存器,用于控制各元素是否参与计算。mask中与元素对应的比特位为1时,该元素参与计算;为0时,该元素不参与计算。

矢量数据寄存器和掩码寄存器的详细说明请参见reg数据类型定义

返回值说明

约束说明

  • srcdst的数据类型需要与函数原型匹配。
  • mask掩码位为0时,dst对应元素置0。
  • 结果写入dst的奇数索引位置时,偶数索引位置置0;结果写入偶数索引位置时,奇数索引位置置0。
  • 使用饱和或非饱和模式时,需要配置ctrl寄存器,详细说明请参见asc_set_ctrl

调用示例

C++
vector_bfloat16_t dst;
vector_float src;
vector_bool mask;
mask = asc_create_mask_b32(PAT_ALL);
asc_float2bfloat16_rd(dst, src, mask);    // FLOOR舍入模式,不饱和模式,将src转换成vector_bfloat16_t类型并写入dst偶数索引位置
asc_float2bfloat16_rd_sat(dst, src, mask);    // FLOOR舍入模式,饱和模式,将src转换成vector_bfloat16_t类型并写入dst偶数索引位置
asc_float2bfloat16_rd_v2(dst, src, mask);    // FLOOR舍入模式,不饱和模式,将src转换成vector_bfloat16_t类型并写入dst奇数索引位置
asc_float2bfloat16_rd_sat_v2(dst, src, mask);    // FLOOR舍入模式,饱和模式,将src转换成vector_bfloat16_t类型并写入dst奇数索引位置

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