aclblasCreateHandleForHgemm
Applicability
|
Product |
Supported |
|---|---|
|
Atlas 350 Accelerator Card |
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
Function Usage
Creates a handle to matrix-matrix multiplication. Both the input and output are of data type aclFloat16. After the handle is created, call aclopExecWithHandle to execute the operator.
Prototype
1 2 3 4 5 6 7 8 |
aclError aclblasCreateHandleForHgemm(aclTransType transA, aclTransType transB, aclTransType transC, int m, int n, int k, aclComputeType type, aclopHandle **handle) |
Parameters
|
Parameter |
Input/Output |
Description |
|---|---|---|
|
transA |
Input |
Transpose flag of matrix A. For details about the type definition, see aclTransType. |
|
transB |
Input |
Transpose flag of matrix B. For details about the type definition, see aclTransType. |
|
transC |
Input |
Transpose flag of matrix C. Currently, the value can only be ACL_TRANS_N. For details about the type definition, see aclTransType. |
|
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. |
|
type |
Input |
Computation precision type (high precision by default). For details about the type definition, see aclComputeType. |
|
handle |
Output |
Pointer to the operator execution handle's pointer. For details about the type definition, see aclopHandle. |
Returns
0 on success; else, failure. For details, see aclError.