功能说明
在计算tiling即调用GetTiling接口前,设置表1中的MatmulConfig相关参数,详细MatmulConfig参数请见表2。
函数原型
void SetMatmulConfigParams(int32_t mmConfigTypeIn = 1, bool enableL1CacheUBIn = false)
参数说明
表1 参数说明参数名
|
输入/输出
|
描述
|
mmConfigTypeIn
|
输入
|
设置matmul的模板类型(与matmul对象创建的模板一致),默认值为1;0代表Norm模板,1代表MDL模板,当前只支持配置0或1。
|
enableL1CacheUBIn
|
输入
|
配置是否使能L1缓存UB计算块,默认值为false;仅在Atlas推理系列产品AI Core场景生效。参考使能场景:MTE3和MTE2流水串行较多的场景。
|