aclblasCreateHandleForS8gemm
产品支持情况
产品 |
是否支持 |
|---|---|
Atlas 350 加速卡 |
√ |
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
创建矩阵-矩阵乘的handle,输入数据的数据类型为int8_t,输出数据的数据类型为int32_t。创建handle成功后,需调用aclopExecWithHandle接口执行算子。
函数原型
1 2 3 4 5 6 7 8 | aclError aclblasCreateHandleForS8gemm(aclTransType transA, aclTransType transB, aclTransType transC, int m, int n, int k, aclComputeType type, aclopHandle **handle) |
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
transA |
输入 |
A矩阵是否转置的标记。类型定义请参见aclTransType。 |
transB |
输入 |
B矩阵是否转置的标记。类型定义请参见aclTransType。 |
transC |
输入 |
C矩阵的标记。类型定义请参见aclTransType。 当前仅支持ACL_TRANS_N。 |
m |
输入 |
矩阵A的行数与矩阵C的行数。 |
n |
输入 |
矩阵B的列数与矩阵C的列数。 |
k |
输入 |
矩阵A的列数与矩阵B的行数。 |
type |
输入 |
计算精度,默认高精度。类型定义请参见aclComputeType。 |
handle |
输出 |
“算子执行handle的指针”的指针。类型定义请参见aclopHandle。 |
返回值说明
返回0表示成功,返回其他值表示失败,请参见aclError。
父主题: CBLAS接口