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

2.6.13. cublastpmv()

cublasStatus_t cublasStpmv(cublasHandle_t handle, cublasFillMode_t uplo,      cublasOperation_t trans, cublasDiagType_t diag,      int n, const float    *AP,      float    *x, int incx)cublasStatus_t cublasDtpmv(cublasHandle_t handle, cublasFillMode_t uplo,      cublasOperation_t trans, cublasDiagType_t diag,      int n, const double   *AP,      double   *x, int incx)cublasStatus_t cublasCtpmv(cublasHandle_t handle, cublasFillMode_t uplo,      cublasOperation_t trans, cublasDiagType_t diag,      int n, const cuComplex*AP,      cuComplex*x, int incx)cublasStatus_t cublasZtpmv(cublasHandle_t handle, cublasFillMode_t uplo,      cublasOperation_t trans, cublasDiagType_t diag,      int n, const cuDoubleComplex *AP,      cuDoubleComplex *x, int incx)

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

$x = op(A)x $

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

o p ( A ) ={ A     如果 t r a n s a = = C U B L A S _ O P _ N , A T   如果 t r a n s a = = C U B L A S _ O P _ T , A H   如果 t r a n s a = = C U B L A S _ O P _ H op(A)= \begin{cases} A\ \ \ \ 如果 transa == CUBLAS\_OP\_N,\\ A^T \ \ 如果 transa == CUBLAS\_OP\_T,\\ A^H \ \ 如果 transa == CUBLAS\_OP\_H \end{cases} op(A)=⎩⎨⎧​A    如果transa==CUBLAS_OP_N,AT  如果transa==CUBLAS_OP_T,AH  如果transa==CUBLAS_OP_H​

解决方案 x 在退出时覆盖右侧的 b。

此函数中不包含对奇点或接近奇点的测试。

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

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

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

请参考:

stpmv, dtpmv, ctpmv, ztpmv