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

2.6.4. cublassbmv()

cublasStatus_t cublasSsbmv(cublasHandle_t handle, cublasFillMode_t uplo,
                           int n, int k, const float  *alpha,
                           const float  *A, int lda,
                           const float  *x, int incx,
                           const float  *beta, float *y, int incy)
cublasStatus_t cublasDsbmv(cublasHandle_t handle, cublasFillMode_t uplo,
                           int n, int k, const double *alpha,
                           const double *A, int lda,
                           const double *x, int incx,
                           const double *beta, double *y, int incy)
1
2
3
4
5
6
7
8
9
10

此函数执行对称带状矩阵向量乘法:

$y=\alpha Ax+\beta y$

其中 A 是具有 k 个子对角线和超对角线的 n*n 对称带状矩阵，并且是向量，x 和 y 是标量。

如果 uplo == CUBLAS_FILL_MODE_LOWER 则对称带状矩阵 A 逐列存储，矩阵的主对角线存储在第 1 行，第 2 行的第一个下对角线（从第一个位置开始），第 3 行的第二个下对角线（开始 在第一个位置）等。因此，一般来说，元素 A(i,j) 存储在内存位置 A(1+i-j,j) 中 i=1,…,n 和 $i\in [j,min(m,j+k)]$ 。 此外，在概念上不对应于带状矩阵（右下角 k*k 三角形）中的元素的数组 A 中的元素不被引用。

如果 uplo == CUBLAS_FILL_MODE_UPPER 则对称带状矩阵 A 逐列存储，矩阵的主对角线存储在第 k+1 行，第 k 行的第一个下对角线（从第一个位置开始），第 k-1 行的第二个下对角线（开始 在第一个位置）等。因此，一般来说，元素 A(1+k+i-j,j) 存储在内存位置 A(1+i-j,j) 中 i=1,…,n 和 $i\in [max(1,j-k),j]$ 。 此外，在概念上不对应于带状矩阵（右下角 k*k 三角形）中的元素的数组 A 中的元素不被引用。

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

请参考:

ssbmv, dsbmv