CANN/asc-devkit获取Matmul Tiling参数

GetTiling【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明获取Tiling参数。函数原型int64_t GetTiling(optiling::TCubeTiling tiling)int64_t GetTiling(AscendC::tiling::TCubeTiling tiling)参数说明表1参数说明参数名输入/输出描述tiling输出Tiling结构体存储最终的tiling结果。TCubeTiling结构介绍请参考表1。optiling::TCubeTiling带有optiling命名空间的TCubeTiling结构体该结构体为Host侧定义的Matmul TilingData。AscendC::tiling::TCubeTiling带有AscendC::tiling命名空间的TCubeTiling结构体Kernel侧定义的Matmul TilingData与使用标准C语法定义TilingData结构体的开发方式配合使用。返回值说明如果返回值不为-1则代表Tiling计算成功用户可以使用该Tiling结构的值。如果返回值为-1则代表Tiling计算失败该Tiling结果无法使用。约束说明在Tiling计算失败的场景若需查看Tiling计算失败的原因请将日志级别设置为WARNING级别并在日志中搜索关键字“MatmulApi Tiling”。在Tiling计算成功的场景若需查看Tiling结构体的参数值请将日志级别设置为INFO级别并在日志中搜索关键字“MatmulTiling”。调用示例auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); matmul_tiling::MatmulApiTiling tiling(ascendcPlatform); tiling.SetAType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT16); tiling.SetBType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT16); tiling.SetCType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT); tiling.SetBiasType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT); tiling.SetShape(1024, 1024, 1024); tiling.SetOrgShape(1024, 1024, 1024); tiling.SetBias(true); tiling.SetBufferSpace(-1, -1, -1); optiling::TCubeTiling tilingData; int ret tiling.GetTiling(tilingData); // 获取Tiling参数【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考