产品 |
是否支持 |
|---|---|
Ascend 950PR/Ascend 950DT |
√ |
√ |
|
√ |
|
√ |
|
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精度进行计算。 |