Skip to content

SetSplitRange

功能说明

设置baseM/baseN/baseK的最大值和最小值。目前Tiling暂时不支持该功能。

函数原型

Text
int32_t SetSplitRange(int32_t maxBaseM = -1, int32_t maxBaseN = -1, int32_t maxBaseK = -1, int32_t minBaseM = -1, int32_t minBaseN = -1, int32_t minBaseK = -1)

参数说明

表1 参数说明

参数名输入/输出描述
maxBaseM输入设置最大的baseM值,默认值为-1。-1表示不设置指定的baseM最大值,该值由Tiling函数自行计算。
maxBaseN输入设置最大的baseN值,默认值为-1。-1表示不设置指定的baseN最大值,该值由Tiling函数自行计算。
maxBaseK输入设置最大的baseK值,默认值为-1。-1表示不设置指定的baseK最大值,该值由Tiling函数自行计算。
minBaseM输入设置最小的baseM值,默认值为-1。-1表示不设置指定的baseM最小值,该值由Tiling函数自行计算。
minBaseN输入设置最小的baseN值,默认值为-1。-1表示不设置指定的baseN最小值,该值由Tiling函数自行计算。
minBaseK输入设置最小的baseK值,默认值为-1。-1表示不设置指定的baseK最小值,该值由Tiling函数自行计算。

返回值说明

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

约束说明

若baseM/baseN/baseK不满足C0_size对齐,计算Tiling时会将该值对齐到C0_size。提示,half/bfloat16_t数据类型的C0_size为16,float数据类型的C0_size为8,int8_t数据类型的C0_size为32,int4b_t数据类型的C0_size为64。

免责声明:本站内容由 asc-devkit 仓 master 分支自动编译生成,属于持续开发版本,可能存在缺陷,仅供预览与参考。如需稳定及商用资料,请查阅官方 昇腾社区