Description: Performs matrix multiplication on the tensors self and mat2. mat2 supports only the AI processor affinity data layout format. self must be 3D, and mat2 must be 5D.
Formula:
Each operator has calls. First, aclnnBatchMatMulWeightNzGetWorkspaceSize is called to obtain the input parameters and compute the required workspace size based on the process. Then, aclnnBatchMatMulWeightNZ is called to perform computation.
Parameters:
[object Object]Returns:
aclnnStatus: status code. For details, see .
The first-phase API implements input parameter verification. The following errors may be thrown.
[object Object]
- Determinism:
- [object Object]Atlas training series products[object Object] and [object Object]Atlas inference series products[object Object]: aclnnBatchMatMulWeightNz defaults to a deterministic implementation.
- If one input is BFLOAT16 and the other is FLOAT16, the data type cannot be deduced.
The data types of self and mat2 are float16. The sample code when mat2 is in AI processor affinity format is as follows (for reference only). For details about the compilation and running process, see .