Ctrmv
功能描述
计算公式
接口原型
- AspbStatus asdBlasMakeCtrmvPlan(asdBlasHandle handle, asdBlasFillMode_t uplo, int64_t n)
- AspbStatus asdBlasCtrmv(asdBlasHandle handle, asdBlasFillMode_t uplo, asdBlasOperation_t trans, asdBlasDiagType_t diag,const int64_t n, aclTensor *A, const int64_t lda, aclTensor *x, const int64_t incx)
参数列表
- asdBlasMakeCtrmvPlan
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
ctrmv算子的句柄。
uplo
Input
asdBlasFillMode_t
指定矩阵A的存储格式。
ASDBLAS_FILL_MODE_LOWER // 下三角 ASDBLAS_FILL_MODE_UPPER // 上三角
n
Input
int64_t
矩阵A的阶数,向量x的维度。
- asdBlasCtrmv
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
ctrmv算子的句柄。
uplo
Input
asdBlasFillMode_t
指定矩阵A的存储格式。
ASDBLAS_FILL_MODE_LOWER // 下三角 ASDBLAS_FILL_MODE_UPPER // 上三角
trans
Input
asdBlasDiagType_t
指定矩阵A是否需要转置。
ASDBLAS_OP_N // 不转置 ASDBLAS_OP_T // 转置 ASDBLAS_OP_C // 共轭转置
diag
Input
asdBlasDiagType_t
指定矩阵A的对角元素处理方式。
ASDBLAS_DIAG_NON_UNIT // 一般矩阵 ASDBLAS_DIAG_UNIT // 单位矩阵
n
Input
const int64_t
矩阵A的阶数,向量x的维度。
A
Input
aclTensor *
公式中的A,Device侧的Tensor,数据类型仅支持COMPLEX64,数据格式支持ND,shape为[n,n]。
lda
Input
const int64_t
矩阵A的第一个维度大小。
x
Input/Output
aclTensor *
公式中的x,Device侧的Tensor,数据类型仅支持COMPLEX64,数据格式支持ND,shape为[n]。
incx
Input
const int64_t
向量x中元素的步长。
规格约束
- asdBlasMakeCtrmvPlan:无。
- asdBlasCtrmv
- 输入的元素个数n当前覆盖支持[1,8192];
- 算子输入shape为[n,n]、[n],输出shape为[n];
- 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。
父主题: BLAS