SetFixSplit

功能说明

设置固定的baseM、baseN、baseK,单位为元素个数。

函数原型

1
int32_t SetFixSplit(int32_t baseMIn = -1, int32_t baseNIn = -1, int32_t baseKIn = -1)

参数说明

表1 参数说明

参数名

输入/输出

描述

baseMIn

输入

设置固定的baseM,默认值为-1,表示不设置固定baseM,由tiling函数进行计算。

baseNIn

输入

设置固定的baseN,默认值为-1,表示不设置固定baseN,由tiling函数进行计算。

baseKIn

输入

当前仅支持取值为-1,暂不支持设置其它值。

返回值说明

-1表示设置失败;0表示设置成功。

约束说明

调用示例

1
2
3
4
auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
matmul_tiling::MatmulApiTiling tiling(ascendcPlatform); 

tiling.SetFixSplit(16, 16, -1);  // 设置固定的baseM、baseN