GetTiling
功能说明
获取Tiling参数。
函数原型
1 | int64_t GetTiling(optiling::TCubeTiling &tiling) |
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
tiling |
输出 |
Tiling结构体存储最终的tiling结果。TCubeTiling结构介绍请参考表1。 |
返回值
如果返回值不为-1,则代表Tiling计算成功,用户可以使用该Tiling结构的值。如果返回值为-1,则代表Tiling计算失败,该Tiling结果无法使用。
注意事项
无
调用示例
1 2 3 4 5 6 7 8 9 10 11 12 | 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参数 |
父主题: Matmul Tiling类