aclblasCreateHandleForGemmEx

Applicable Products

Product

Supported

Ascend 950PR / Ascend 950DT

×

Atlas A3 training products / Atlas A3 inference products

√

Atlas A2 training products / Atlas A2 inference products

√

Atlas 200I/500 A2 inference products

√

Atlas inference products

√

Atlas training products

√

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

FLOAT16

FLOAT16

FLOAT16

FLOAT16

FLOAT16

FLOAT

INT8

INT8

FLOAT

INT8

INT8

INT32

Prototype

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
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 the type definition, see aclDataType.

dataTypeB

Input

Data type of matrix B. For the type definition, see aclDataType.

dataTypeC

Input

Data type of matrix C. For 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.