设置固定的baseM、baseN、baseK,单位为元素个数。
1 | int32_t SetFixSplit(int32_t baseMIn = -1, int32_t baseNIn = -1, int32_t baseKIn = -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 |