aclblasCreateHandleForGemmEx

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product / Atlas A3 inference product	√
Atlas A2 training product / Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Function Usage

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
aclFloat16	aclFloat16	aclFloat16
aclFloat16	aclFloat16	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. 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. For details about the type definition, see aclTransType. Currently, the value can only be ACL_TRANS_N.
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. For details about the type definition, see aclDataType.
dataTypeB	Input	Data type of matrix B. For details about the type definition, see aclDataType.
dataTypeC	Input	Data type of matrix C. For details about the type definition, see aclDataType.
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.

Parent topic: CBLAS APIs