SetHF32Mode
产品支持情况
产品 |
是否支持 |
|---|---|
√ |
|
√ |
|
√ |
|
x |
|
x |
|
x |
功能说明
用于设置Mmad计算是否开启HF32模式，开启该模式后L0A/L0B中的FP32数据将在参与Mmad计算之前被舍入为HF32。
函数原型
1 | __aicore__ inline void SetHF32Mode(HF32Mode mode) |
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
mode |
输入 |
Mmad HF32模式控制入参，HF32Mode枚举类型。支持如下两种取值：
|
返回值说明
无
约束说明
无
调用示例
1 | AscendC::SetHF32Mode(HF32Mode::ENABLE); // 控制mmad计算时是否使用HF32精度进行计算。 |
父主题： 矩阵计算