文档
注册
评分
提单
论坛
小AI

GetBaseM

功能说明

获取Tiling计算得到的baseM值。

函数原型

1
int32_t GetBaseM()

参数说明

返回值

返回值为Tiling计算得到的baseM值。

注意事项

需使用创建的Tiling对象去调用该接口,且需在完成Tiling计算(GetTiling)后调用。

调用示例

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);

int baseM = tiling.GetBaseM();  // 获取tiling计算得到的baseM
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词