Function: create_handle_for_gemv_ex

Applicability

Product

Supported (√/x)

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas training products

Atlas inference products

Atlas 200I/500 A2 inference products

Function Usage

Creates a handle to matrix-vector multiplication. The data types of the input and output data are set using arguments.

After the handle is created, call acl.op.execute_with_handle to execute the operator.

Prototype

  • C Prototype
    1
    2
    3
    4
    5
    6
    7
    8
    aclError aclblasCreateHandleForGemvEx(aclTransType transA,
    int m,
    int n,
    aclDataType dataTypeA,
    aclDataType dataTypeX,
    aclDataType dataTypeY,
    aclComputeType type,
    aclopHandle **handle)
    
  • Python Function
    1
    handle, ret = acl.blas.create_handle_for_gemv_ex(trans_a, m, n, data_type_a, data_type_x, data_type_y, type)
    

Parameter Description

Parameter

Description

trans_a

Int, whether matrix A is transposed.

m

Int, number of rows in matrix A (row-major first in data storing).

n

Int, number of columns in matrix A.

data_type_a

Int, data type of matrix A.

data_type_x

Int, data type of vector x.

data_type_y

Int, data type of vector y.

type

Int, computation precision.

Return Value Description

Return Value

Description

handle

Int, pointer address of the handle data of the operator to be executed.

ret

Int, error code: 0 on success; else, failure.

Restrictions

A, x, and y support a limited combination of data types, where α and β have the same data types as y.

Data Type of A

Data Type of x

Data Type of y

aclFloat16

aclFloat16

aclFloat16

aclFloat16

aclFloat16

float(float32)

int8_t

int8_t

float(float32)

int8_t

int8_t

int32_t