Matmul Feature Scenarios
In addition to the basic computing capabilities of Matmul described above, you also need to master the basic knowledge and operator implementation of Matmul. In addition, Matmul matrix programming provides processing capabilities and multiple functions applicable to different scenarios. The following table lists the specific scenarios and key APIs or parameters used. For details, see the corresponding sections.
|
Feature Description |
Key API or Parameter |
|---|---|
|
SetDim and EnableMultiCoreSplitK (multi-core K splitting) |
|
|
SetTail, SetDim, and EnableMultiCoreSplitK (multi-core K splitting) |
|
|
Iterate, GetTensorC, and IterateAll |
|
|
Quantization and dequantization of matrix multiplication outputs |
|
|
Upper/Lower triangular matrix multiplication |
MatmulPolicy template parameters |
|
SetScaleAType, SetScaleBType, SetMadType, SetTensorScaleA, and SetTensorScaleB |
|
Feature Description |
Involved API |
|---|---|
|
BatchMatmul in the NORMAL format: IterateBatch and SetBatchInfoForNormal |
|
|
BatchMatmul in the BSNGD, SBNGD, and BNGS1S2 formats: IterateBatch, SetALayout, SetBLayout, SetCLayout, and SetBatchNum |
|