接口 |
功能 |
|---|---|
Init |
初始化Matmul对象，传入tiling参数。 |
SetTensorA |
设置矩阵乘的左矩阵A。 |
SetTensorB |
设置矩阵乘的右矩阵B。 |
SetBias |
设置矩阵乘的Bias。 |
ClearBias |
清除Bias标志位，表示matmul计算时没有Bias参与。 |
Iterate |
每调用一次Iterate，会计算出一片baseM * baseN的C矩阵。 |
IterateAll |
调用IterateAll，会计算出singleCoreM * singleCoreN大小的C矩阵。 |
End |
一个matmul计算结束时需要调用一次End函数。 |
SetHF32 |
设置是否使能HF32模式。当前版本暂不支持。 |
GetOffsetC |
获取本次计算时当前分片在整个C矩阵中的位置。 |
SetTail |
在不改变tiling的场景下，重新设置本次计算的singleM/singleN/singleK。 |
GetTensorC |
配合Iterate使用，一次Iterate后，获取一块C矩阵片。 |