函数:create_handle_for_gemm_ex
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) |
函数功能 |
创建矩阵-矩阵乘的handle,输入数据、输出数据的数据类型通过入参设置。 创建handle成功后,需调用acl.op.execute_with_handle接口执行算子。 |
输入说明 |
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,错误码。
|
约束说明 |
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 |