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