接口 |
功能 |
|---|---|
SetAType |
设置A矩阵的位置，数据格式，数据类型，是否转置等信息。 |
SetBType |
设置B矩阵的位置，数据格式，数据类型，是否转置等信息。 |
SetCType |
设置C矩阵的位置，数据格式，数据类型等信息。 |
SetBiasType |
设置Bias的位置，数据格式，数据类型等信息。 |
SetShape |
设置Matmul单次计算的形状singleM、singleN、singleK，单位为元素个数。 |
SetSingleShape |
设置Matmul单核计算的形状singleCoreM、singleCoreN、singleCoreK，单位为元素个数。 |
SetOrgShape |
设置matmul计算时的原始完整的形状M、N、K，单位为元素个数。 |
SetBias |
设置Bias是否参与运算。 |
SetFixSplit |
设置固定的baseM、baseN、baseK，单位为元素个数。 |
SetBufferSpace |
设置matmul计算时可用的L1/L0C/UB空间大小，单位为字节。 |
SetTraverse |
设置遍历方式，M轴优先还是N轴优先。 |
SetMadType |
设置是否使能HF32模式。当前版本暂不支持。 |
SetSplitRange |
设置baseM/baseN/baseK的最大值和最小值。 |
SetDoubleBuffer |
设置A/B/C/Bias是否使能double buffer功能，以及是否需要做ND2NZ或者NZ2ND的转换。 |
GetBaseM |
获取baseM值。 |
GetBaseN |
获取baseN值。 |
GetBaseK |
获取baseK值。 |
GetTiling |
获取Tiling参数。 |
接口 |
功能 |
|---|---|
SetDim |
设置多核Matmul时，可以参与运算的核数。 |
SetSingleRange |
设置singleCoreM/singleCoreN/singleCoreK的最大值与最小值。 |
GetSingleShape |
获取计算后的singleCoreM/singleCoreN/singleCoreK。 |
GetCoreNum |
获得多核切分后， 使用的BlockDim。 |