昇腾社区首页
中文
注册

函数:gemv_ex

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas 800I A2 推理产品 /A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

执行矩阵-向量的乘法,y = αAx + βy,输入数据、输出数据的数据类型通过入参设置,异步接口。

函数原型

  • C函数原型
     1
     2
     3
     4
     5
     6
     7
     8
     9
    10
    11
    12
    13
    14
    15
    16
    aclError aclblasGemvEx(aclTransType transA,
    int m,
    int n,
    const void *alpha,
    const void *a,
    int lda,
    aclDataType dataTypeA,
    const void *x,
    int incx,
    aclDataType dataTypeX,
    const void *beta,
    void *y,
    int incy,
    aclDataType dataTypeY,
    aclComputeType type,
    aclrtStream stream);
    
  • python函数
    1
    ret = acl.blas.gemv_ex(trans_a, m, n,alpha, a, lda, data_type_a, x, incx, data_type_x, beta, y, incy, data_type_y, type, stream )
    

参数说明

参数名

说明

trans_a

int,矩阵A是否转置的标记。

m

int,矩阵A的行数,存储矩阵乘数据时,行优先。

n

int,矩阵A的列数。

alpha

int, 用于执行乘操作的标量alpha数据指针地址。

a

int,矩阵A的数据指针地址。

lda

int,A矩阵的主位,此时选择转置,按行优先,则lda为A的列数。预留参数,当前只能设置为-1。

data_type_a

int,矩阵A的数据类型。

x

int,向量x的数据指针地址。

incx

int,x连续元素之间的步长。预留参数,当前只能设置为-1。

data_type_x

int,向量x的数据类型。

beta

int,用于执行乘操作的标量beta的数据指针地址。

y

int,向量y的数据指针地址。

incy

int,y连续元素之间的步长。预留参数,当前只能设置为-1。

data_type_y

int,向量y的数据类型。

type

int,计算精度,默认高精度。

stream

int,执行算子所在的Stream。

返回值说明

返回值

说明

ret

int,错误码,返回0表示成功,返回其它值表示失败。

约束说明

A、x、y的数据类型支持仅支持以下组合, α和β的数据类型与y一致。

A的数据类型

x的数据类型

y的数据类型

aclFloat16

aclFloat16

aclFloat16

aclFloat16

aclFloat16

float(float32)

int8_t

int8_t

float(float32)

int8_t

int8_t

int32_t

资源参考

接口调用流程,参见接口调用流程