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