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

2.6.18. cublashbmv()

在这里插入图片描述

cublasStatus_t cublasChbmv(cublasHandle_t handle, cublasFillMode_t uplo,
                          int n, int k, const cuComplex       *alpha,
                          const cuComplex       *A, int lda,
                          const cuComplex       *x, int incx,
                          const cuComplex       *beta,
                          cuComplex       *y, int incy)
cublasStatus_t cublasZhbmv(cublasHandle_t handle, cublasFillMode_t uplo,
                          int n, int k, const cuDoubleComplex *alpha,
                          const cuDoubleComplex *A, int lda,
                          const cuDoubleComplex *x, int incx,
                          const cuDoubleComplex *beta,
                          cuDoubleComplex *y, int incy)

y = α A x + β y y = \alpha Ax+\beta y y=αAx+βy

其中 A 是具有 k 个下对角线和上对角线的 nxn Hermitian 带状矩阵,x 和 y 是向量,α 和 β 是标量。

如果 uplo == CUBLAS_FILL_MODE_LOWER 则厄尔米特带状矩阵 A 逐列存储,矩阵的主对角线存储在第 1 行,第一个次对角线存储在第 2 行(从第一个位置开始),第二个次对角线存储在第 3 行(从第一个位置开始) 在第一个位置)等。所以一般来说,元素 A(i,j) 存储在内存位置 A(1+i-j,j) for j=1,…,n and i ∈ [ j , m i n ( m , j + k ) ] i\in[j, min(m, j+k)] i[j,min(m,j+k)] 。 此外,数组 A 中在概念上与带状矩阵(右下角的 kxk 三角形)中的元素不对应的元素未被引用。

如果 uplo == CUBLAS_FILL_MODE_UPPER 则厄尔米特带状矩阵 A 逐列存储,矩阵的主对角线存储在第 k+1 行,第一个超对角线存储在第 k 行(从第二个位置开始),第二个超对角线存储在第 k 行 -1(从第三个位置开始)等。所以一般来说,元素 A(i,j) 存储在内存位置 A(1+k+i-j,j) 中,j=1,…n 并且 我> 1。 此外,数组 A 中在概念上与带状矩阵(左上角的 k x k 三角形)中的元素不对应的元素未被引用。

Param.MemoryIn/outMeaning
handleinputhandle to the cuBLAS library context.
uploinputIndicates if matrix A lower or upper part is stored, the other Hermitian part is not referenced and is inferred from the stored elements.
ninputnumber of columns of matrix A.
kinputNumber of sub- and super-diagonals of matrix A.
alphahost or deviceinput<type> scalar used for multiplication.
Adeviceinput<type> array of dimension lda x n with lda>=kl+ku+1.
ldainputleading dimension of two-dimensional array used to store matrix A.
xdeviceinput<type> vector with n elements if transa == CUBLAS_OP_N and m elements otherwise.
incxinputstride between consecutive elements of x.
betahost or deviceinput<type> scalar used for multiplication, if beta == 0 then y does not have to be a valid input.
ydevicein/out<type> vector at least (1+(m-1)*abs(incy)) elements if transa==CUBLAS_OP_N and at least (1+(n-1)*abs(incy)) elements otherwise.
incyinputstride between consecutive elements of y.

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

ErrorValueMeaning
CUBLAS_STATUS_SUCCESS操作成功完成
CUBLAS_STATUS_NOT_INITIALIZED库未初始化
CUBLAS_STATUS_INVALID_VALUEIf n < 0 or k < 0 or if incx = 0 or incy = 0 or if uplo != CUBLAS_FILL_MODE_LOWER, CUBLAS_FILL_MODE_UPPER or if lda < (k + 1) or alpha == NULL or beta == NULL
CUBLAS_STATUS_EXECUTION_FAILED该功能无法在 GPU 上启动

详细信息, 请参考:

chbmv, zhbmv

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

扫地的小何尚

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值