函数：create_handle_for_gemm_ex

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 训练系列产品	√
Atlas 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 200/300/500 推理产品	√

功能说明

创建矩阵-矩阵乘的handle，输入数据、输出数据的数据类型通过入参设置。

创建handle成功后，需调用acl.op.execute_with_handle接口执行算子。

函数原型

C函数原型

aclError aclblasCreateHandleForGemmEx(aclTransType transA,
aclTransType transB,
aclTransType transC,
int m,
int n,
int k,
aclDataType dataTypeA,
aclDataType dataTypeB,
aclDataType dataTypeC,
aclComputeType type,
aclopHandle **handle)

python函数

handle, ret = acl.blas.create_handle_for_gemm_ex(trans_a, trans_b, trans_c, m, n, k, data_type_a, data_type_b, data_type_c,type)

参数说明

参数名	说明
trans_a	int，矩阵A是否转置的标记。
trans_b	int，B矩阵是否转置的标记。
trans_c	int，C矩阵的标记，当前仅支持aclTrans_N。
m	int，矩阵A的行数与矩阵C的行数。
n	int，矩阵B的列数与矩阵C的列数。
k	int，矩阵A的列数与矩阵B的行数。
data_type_a	int，矩阵A的数据类型。
data_type_b	int，矩阵B的数据类型。
data_type_c	int，矩阵C的数据类型。
type	int，计算精度。

返回值说明

返回值	说明
handle	int，执行算子的handle数据的指针地址。
ret	int，错误码，返回0表示成功，返回其它值表示失败。

约束说明

A、B、C的数据类型支持仅支持以下组合：

A的数据类型	x的数据类型	y的数据类型
aclFloat16	aclFloat16	aclFloat16
aclFloat16	aclFloat16	float(float32)
int8_t	int8_t	float(float32)
int8_t	int8_t	int32_t

父主题： CBLAS接口（blas）