Matmul特性场景
除了前述介绍的Matmul基本计算能力外,还请掌握Matmul的基础知识和算子实现。另外,Matmul矩阵编程还提供了适用于不同场景的处理能力及多种功能,具体场景和使用的关键接口或参数列于下表中,详细内容请见对应章节的介绍。
特性分类 |
特性描述 |
涉及的关键API或参数 |
---|---|---|
功能实现 |
SetDim、EnableMultiCoreSplitK(多核切K场景) |
|
SetTail、SetDim、EnableMultiCoreSplitK(多核切K场景) |
||
MatmulConfig模板参数中的isEnableChannelSplit参数 |
||
MatmulPolicy模板参数 |
||
MatmulConfig模板参数中的isPartialOutput参数 |
||
MatmulConfig模板参数中的enableMixDualMaster参数 |
特性分类 |
特性描述 |
主要涉及的API接口 |
---|---|---|
功能实现 |
BatchMatmul基础场景 |
NORMAL排布格式的BatchMatmul:IterateBatch、SetBatchInfoForNormal |
BSNGD、SBNGD、BNGS1S2排布格式的BatchMatmul:IterateBatch、SetALayout、SetBLayout、SetCLayout、SetBatchNum |
父主题: Matmul