SetMMLayoutTransform
功能说明
此接口同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 200/500 A2推理产品
约束说明
无。
调用示例
1 2 |
bool hf32Mode = true; AscendC::SetHF32Mode(hf32Mode); |
父主题: 矩阵计算