CANN/asc-devkit单核形状获取

GetSingleShape【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明获取计算后的singleCoreM/singleCoreN/singleCoreK。函数原型int32_t GetSingleShape(int32_t shapeM, int32_t shapeN, int32_t shapeK)参数说明表1参数说明参数名输入/输出描述shapeM输出获取多核Tiling计算得到的singleCoreM值shapeN输出获取多核Tiling计算得到的singleCoreN值shapeK输出获取多核Tiling计算得到的singleCoreK值返回值说明-1表示设置失败 0表示设置成功。约束说明使用创建的Tiling对象调用该接口且需在完成Tiling计算GetTiling后调用。调用示例auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); matmul_tiling::MultiCoreMatmulTiling tiling(ascendcPlatform); tiling.SetDim(1); 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.SetSingleShape(1024, 1024, 1024); tiling.SetOrgShape(1024, 1024, 1024); tiling.SetBias(true); tiling.SetBufferSpace(-1, -1, -1); optiling::TCubeTiling tilingData; int ret tiling.GetTiling(tilingData); // 获取计算后的singleCoreM/singleCoreN/singleCoreK int32_t singleM, singleN, singleK; int ret1 tiling.GetSingleShape(singleM, singleN, singleK);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考