SetHF32

功能说明

设置是否使能HF32模式。

函数原型

      
           __aicore__ inline void SetHF32(bool enableHF32 = false, int32_t transMode = 0)

参数说明

参数名	输入/输出	描述
enableHF32	输入	配置是否开启HF32模式，默认值false(不开启)。
transMode	输入	配置在开启HF32模式时，float32转换为hf32时所采用的ROUND模式。默认值0。 0: round mode is round to the nearest tie to even。 1: round mode is round to the nearest tie away from zero

参数名

输入/输出

描述

enableHF32

输入

配置是否开启HF32模式，默认值false(不开启)。

transMode

输入

配置在开启HF32模式时，float32转换为hf32时所采用的ROUND模式。默认值0。

0: round mode is round to the nearest tie to even。

1: round mode is round to the nearest tie away from zero

返回值

无

支持的型号

Atlas A2训练系列产品/Atlas 800I A2推理产品

注意事项

无

调用示例

REGIST_MATMUL_OBJ(&pipe, GetSysWorkSpacePtr(), mm, &tiling);    //  A/B/C/BIAS类型是float
mm.SetTensorA(gm_a);
mm.SetTensorB(gm_b);
mm.SetBias(gm_bias);
mm.SetHF32(true);
mm.IterateAll(gm_c);

父主题： Matmul