GetBaseK

Function

Obtains the baseK value generated by the tiling computation. For details about the baseK parameter, see Table 1.

Prototype

int32_t GetBaseK() const

Parameters

None

Returns

The return value is the baseK value generated by the tiling computation.

Restrictions

This API needs to be called using the created tiling object after the tiling computation (GetTiling) is complete.

Example

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 baseK = tiling.GetBaseK();  // Obtain baseK generated by the tiling computation.

Parent topic: Matmul Tiling