cuBLAS中的Level-1函数rotm()和rotmg()

2.5.10. cublasrotm()

cublasStatus_t cublasSrotm(cublasHandle_t handle, int n, float  *x, int incx,
                           float  *y, int incy, const float*  param)
cublasStatus_t cublasDrotm(cublasHandle_t handle, int n, double *x, int incx,
                           double *y, int incy, const double* param)
1
2
3
4

此函数应用修改后的 Givens 变换

H = h 1 1 h 1 2 h 2 1 h 2 2

到向量 x 和 y。

因此，结果是 x [ k ] = h 1 1 × x [ k ] + h 1 2 × y [ j ] 和 y [ j ] = h 2 1 × x [ k ] + h 2 2 × y [ j ] 其中 k = 1 + ( i - 1 ) * incx 和 j = 1 + ( i - 1 ) * incy 。 请注意，最后两个等式反映了用于与 Fortran 兼容的基于 1 的索引。

矩阵 H 的元素 、 和 分别存储在 param[1]、param[2]、param[3] 和 param[4] 中。 flag=param[0] 为矩阵 H 项定义以下预定义值

请注意，标志所隐含的值 -1.0、0.0 和 1.0 未存储在 param 中。

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

请参考:
srotm, drotm

2.5.11. cublasrotmg()

cublasStatus_t cublasSrotmg(cublasHandle_t handle, float  *d1, float  *d2,
                            float  *x1, const float  *y1, float  *param)
cublasStatus_t cublasDrotmg(cublasHandle_t handle, double *d1, double *d2,
                            double *x1, const double *y1, double *param)
1
2
3
4

此函数构造修改后的 Givens 变换

H = h 1 1 h 1 2 h 2 1 h 2 2

将 2 × 1 向量 (d 1 * x 1 , d 2 * y 1 ) T 的第二个条目归零。

flag=param[0] 为矩阵 H 项定义以下预定义值

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

请参考:
srotmg, drotmg