SetTraverse
【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit
功能说明
设置固定的Matmul计算方向,M轴优先还是N轴优先。
函数原型
int32_t SetTraverse(MatrixTraverse traverse)参数说明
表 1参数说明
设置固定的Matmul计算方向。可选值:MatrixTraverse::FIRSTM/MatrixTraverse::FIRSTN。 |
返回值说明
-1表示设置失败;0表示设置成功。
约束说明
无
调用示例
auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); matmul_tiling::MatmulApiTiling tiling(ascendcPlatform); tiling.SetTraverse(MatrixTraverse::FIRSTM); // 设置遍历方式【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考