SetTraverse
功能说明
设置固定的Matmul计算方向，M轴优先还是N轴优先。
函数原型
Text
int32_t SetTraverse(MatrixTraverse traverse)
参数说明
表1 参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| traverse | 输入 | 设置固定的Matmul计算方向。可选值：MatrixTraverse::FIRSTM/MatrixTraverse::FIRSTN。 FIRSTM代表先往M轴方向偏移再往N轴方向偏移。 FIRSTN代表先往N轴方向偏移再往M轴方向偏移。 |
返回值说明
-1表示设置失败；0表示设置成功。
约束说明
无
调用示例
Text
auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
matmul_tiling::MatmulApiTiling tiling(ascendcPlatform);
tiling.SetTraverse(MatrixTraverse::FIRSTM); // 设置遍历方式