aclblasCreateHandleForGemmEx
Applicability
Product |
Supported |
|---|---|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
Description
Creates a handle to matrix-matrix multiplication. The data types of the input and output data are set using arguments. After the handle is created, call aclopExecWithHandle to execute the operator.
A, B, and C support a limited combination of data types.
Data Type of A |
Data Type of B |
Data Type of C |
|---|---|---|
float(float32) |
||
int8_t |
int8_t |
float(float32) |
int8_t |
int8_t |
int32_t |
Prototype
aclError aclblasCreateHandleForGemmEx(aclTransType transA, aclTransType transB, aclTransType transC, int m, int n, int k, aclDataType dataTypeA, aclDataType dataTypeB, aclDataType dataTypeC, aclComputeType type, aclopHandle **handle)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
transA |
Input |
Transpose flag of matrix A. |
transB |
Input |
Transpose flag of matrix B. |
transC |
Input |
Transpose flag of matrix C. Currently, only ACL_TRANS_N is supported. |
m |
Input |
Number of rows in matrix A and matrix C. |
n |
Input |
Number of columns in matrix B and matrix C. |
k |
Input |
Number of columns in matrix A and number of rows in matrix B. |
dataTypeA |
Input |
Data type of matrix A. |
dataTypeB |
Input |
Data type of matrix B. |
dataTypeC |
Input |
Data type of matrix C. |
type |
Input |
Computation precision type (high precision by default). |
handle |
Output |
Pointer to the pointer to the operator handle. |
Returns
0 on success; else, failure. For details, see aclError.