aclblasCreateHandleForHgemm

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. 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.