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的显式位置参数重载替代。
根据mask将src中的每个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 |
_v2 | ASC_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>表示支持的舍入模式,支持rd(FLOOR)、rn(RINT)、rna(ROUND)、ru(CEIL)和rz(TRUNC)。<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数据类型定义。
返回值说明
无
约束说明
src与dst的数据类型需要与函数原型匹配。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奇数索引位置