Description: Implements grouped matrix multiplication, supporting non-uniform matrix dimension sizes across multiple groups. The basic function is matrix multiplication, for example, , where indicates the number of groups and , , and define the shapes for each group. The input and output data types are both aclTensor, with K-axis grouping.
- K-axis grouping: can vary, while and remain the same for each group.
Formula:
Each operator has calls. First, [object Object] is called to obtain the input parameters and compute the required workspace size based on the process. Then, [object Object] is called to perform computation.
Parameters
[object Object]Ascend 950PR/Ascend 950DT:
- The superscript "1" in the dimension column of the preceding table indicates that the data type is not supported by this series.
Return
[object Object]: status code. For details, see .The first-phase API implements input parameter validation. The following errors may be thrown:
[object Object]
[object Object]defaults to a deterministic implementation.- The size of each dimension for every tensor in
[object Object]and[object Object], after 32-byte alignment, should be less than the maximum value of INT32 (2147483647). - Supported input types:
[object Object]: FLOAT16;[object Object]: FLOAT16;[object Object]: FLOAT32[object Object]: BFLOAT16;[object Object]: BFLOAT16;[object Object]: FLOAT32
The following example is for reference only. For details, see .