32.cuBLAS开发指南中文版--cuBLAS中的Level-2函数trmv()

2.6.15. cublas<t>trmv()

cublasStatus_t cublasStrmv(cublasHandle_t handle, cublasFillMode_t uplo,
                           cublasOperation_t trans, cublasDiagType_t diag,
                           int n, const float           *A, int lda,
                           float           *x, int incx)
cublasStatus_t cublasDtrmv(cublasHandle_t handle, cublasFillMode_t uplo,
                           cublasOperation_t trans, cublasDiagType_t diag,
                           int n, const double          *A, int lda,
                           double          *x, int incx)
cublasStatus_t cublasCtrmv(cublasHandle_t handle, cublasFillMode_t uplo,
                           cublasOperation_t trans, cublasDiagType_t diag,
                           int n, const cuComplex       *A, int lda,
                           cuComplex       *x, int incx)
cublasStatus_t cublasZtrmv(cublasHandle_t handle, cublasFillMode_t uplo,
                           cublasOperation_t trans, cublasDiagType_t diag,
                           int n, const cuDoubleComplex *A, int lda,
                           cuDoubleComplex *x, int incx)

此函数执行三角带状矩阵向量乘法

$x=op(A)x$

其中 A 是三角带状矩阵，x 是向量。 此外，对于矩阵 A

$$op(A)=\{\begin{array}{l}A如果transa==CUBLAS\_OP\_N,\\ A^T如果transa==CUBLAS\_OP\_T,\\ A^H如果transa==CUBLAS\_OP\_H\end{array}$$

如果 uplo == CUBLAS_FILL_MODE_LOWER 则将三角矩阵下三角部分的元素 A(i, j) 逐列无间隙地打包在一起，从而将元素存储在内存位置 AP[i+((2*n -j+1)*j)/2] 对于 j=1 , …,n 和 i>=j 。 因此，打包格式只需要 $\frac{n(n+1)}{2}$ 个元素进行存储。

如果 uplo == CUBLAS_FILL_MODE_UPPER 则将三角矩阵下三角部分的元素 A(i, j) 逐列无间隙地打包在一起，从而将元素存储在内存位置 AP[i+(j*(j+1))/2] 对于 j=1 , …,n 和 i<=j 。 因此，打包格式只需要 $\frac{n(n+1)}{2}$ 个元素进行存储。

Param.	Memory	In/out	Meaning
handle		input	handle to the cuBLAS library context.
uplo		input	indicates if matrix A lower or upper part is stored, the other symmetric part is not referenced and is inferred from the stored elements.
trans		input	operation op(A) that is non- or (conj.) transpose.
diag		input	indicates if the elements on the main diagonal of matrix A are unity and should not be accessed.
n		input	number of rows and columns of matrix A.
A	device	input	<type> array of dimensions lda x n , with lda>=max(1,n).
lda		input	leading dimension of two-dimensional array used to store matrix A.
x	device	input	<type> vector with n elements.
incx		input	stride between consecutive elements of x.

该函数可能返回的错误值及其含义如下所列。

ErrorValue	Meaning
CUBLAS_STATUS_SUCCESS	操作成功完成
CUBLAS_STATUS_NOT_INITIALIZED	库未初始化
CUBLAS_STATUS_ALLOC_FAILED	内部暂存内存分配失败
CUBLAS_STATUS_EXECUTION_FAILED	该功能无法在 GPU 上启动

请参考:

strmv, dtrmv, ctrmv, ztrmv