文档
注册

SetHF32Mode

功能说明

此接口同SetHF32TransModeSetMMLayoutTransform一样,都用于设置寄存器的值。SetHF32Mode接口用于设置MMAD的HF32模式。

函数原型

__aicore__ inline void SetHF32Mode(bool hf32Mode);

参数说明

表1 参数说明

参数名

输入/输出

描述

hf32Mode

输入

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

  • true:则L0A/L0B中的FP32数据将在矩阵乘法之前被CUBE舍入为HF32。
  • false:CUBE将执行常规的FP32矩阵乘法。

返回值

支持的型号

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

约束说明

无。

调用示例

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

当前产品无相关内容

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