aclblasCreateHandleForGemvEx
产品支持情况
产品
|
是否支持
|
|---|
Atlas A3 训练系列产品/Atlas A3 推理系列产品
|
√
|
Atlas A2 训练系列产品/Atlas A2 推理系列产品
|
√
|
功能说明
创建矩阵-向量乘的handle，输入数据、输出数据的数据类型通过入参设置。创建handle成功后，需调用aclopExecWithHandle接口执行算子。
A、x、y的数据类型仅支持以下组合：
A的数据类型
|
x的数据类型
|
y的数据类型
|
|---|
aclFloat16
|
aclFloat16
|
aclFloat16
|
aclFloat16
|
aclFloat16
|
float(float32)
|
int8_t
|
int8_t
|
float(float32)
|
int8_t
|
int8_t
|
int32_t
|
函数原型
aclError aclblasCreateHandleForGemvEx(aclTransType transA,
int m,
int n,
aclDataType dataTypeA,
aclDataType dataTypeX,
aclDataType dataTypeY,
aclComputeType type,
aclopHandle **handle)
参数说明
参数名
|
输入/输出
|
说明
|
|---|
transA
|
输入
|
A矩阵是否转置的标记。
|
m
|
输入
|
矩阵A的行数，存储矩阵乘数据时，行优先。
|
n
|
输入
|
矩阵A的列数。
|
dataTypeA
|
输入
|
矩阵A的数据类型。
|
dataTypeX
|
输入
|
向量x的数据类型。
|
dataTypeY
|
输入
|
向量y的数据类型。
|
type
|
输入
|
计算精度，默认高精度。
|
handle
|
输出
|
“算子执行handle的指针”的指针。
|
返回值说明
返回0表示成功，返回其他值表示失败，请参见aclError。