此接口同SetHF32Mode与SetHF32TransMode一样,都用于设置寄存器的值,其中SetMMLayoutTransform接口用于设置MMAD的M/N方向。
1 | __aicore__ inline void SetMMLayoutTransform(bool mmLayoutMode) |
参数名 |
输入/输出 |
描述 |
---|---|---|
mmLayoutMode |
输入 |
MMAD/MMAD_SP M/N方向控制入参,Bool型,支持如下两种取值:
|
无
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 200I/500 A2推理产品
无。
1 2 | bool hf32Mode = true; AscendC::SetHF32Mode(hf32Mode); |