文档
注册

SetHF32TransMode

功能说明

此接口同SetHF32ModeSetMMLayoutTransform一样,都用于设置寄存器的值。SetHF32TransMode用于设置MMAD的HF32取整模式,仅在MMAD的HF32模式生效时有效。

函数原型

__aicore__ inline void SetHF32TransMode(bool hf32TransMode);

参数说明

表1 参数说明

参数名

输入/输出

描述

hf32TransMode

输入

MMAD HF32取整模式控制入参,bool类型。支持如下两种取值:

  • true:则FP32将以向零靠近的方式四舍五入为HF32。
  • false:则FP32将以最接近偶数的方式四舍五入为HF32。

返回值

支持的型号

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

约束说明

无。

调用示例

bool hf32TransMode = true;
SetHF32TransMode(hf32TransMode);
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词