使用 cblas_chpr 函数

Using the cblas_chpr function

当我尝试使用 cblas 库中的 cblas_chpr() 函数计算浮点复向量的相关矩阵时遇到问题。

从netLib.org下载Lapack v3.10.0 Library后,我编译并复制了libcblas.aliblapack.aliblapacke.alibrefblas.alibtmglib.a 个文件到我的项目并确保库链接正确。

根据描述,cblas_chpr函数计算alpha * x * conjg(x') + A并将结果存入A。

函数定义为:

void cblas_chpr(CBLAS_LAYOUT layout, CBLAS_UPLO Uplo,
                const CBLAS_INDEX N, const float alpha, const void *X,
                const CBLAS_INDEX incX, void *A);

其中参数为:

我的函数体如下:

   /* Number of elements */
   int Ne = 10;


   /* Set the parameters */
   CBLAS_LAYOUT layout = CblasColMajor;   /* Layout is column major */
   CBLAS_UPLO Uplo = CblasUpper;          /* Upper triangle of the matrix */
   CBLAS_INDEX N = Ne;                    /* Number of elements in vector X */
   float alpha = 1.0;                     /* No scaling, alpha = 1.0 */

   /* The vector X */
   float complex * X = malloc(Ne * sizeof(* X));

   /* Set values of X - for illustration purpose only */
   for(int i = 0; i < Ne; i++)
   {
      X[i] = CMPLXF(i, i + 1.0);
   }

   CBLAS_INDEX incX = 1;                  /* Use data from every element */

   /* The correlation matrix is a Ne x Ne matrix */
   float complex ** A = malloc(Ne * sizeof(*A));

   for(int i = 0; i < Ne; i++)
   {
      A[i] = malloc(Ne * sizeof(*A[i]));
   }

   cblas_chpr(layout, Uplo, N, alpha, X, incX, A);

   float complex print_val = A[0][0];
   printf("%+.10f %+.10f", crealf(print_val), cimagf(print_val));

但是,程序崩溃并出现 “chpr_() at 0x55555555e70b”没有可用的源代码 错误。

我猜我的输入参数不正确。 CBLAS 是 Fortran BLAS 库的包装器。

有没有人以前遇到过这个错误并且知道如何解决?

回答我自己的问题,以防其他人 运行 遇到同样的问题。 A 应该是大小为 N * (N + 1) / 2 的一维数组。此外,数组 A 中每个元素的值必须初始化为零。否则,结果将是错误的。阅读 cblas_chpr() 函数的说明,了解为什么会出现这种情况。

/* Number of elements */
int Ne = 10;

/* Set the parameters */
CBLAS_LAYOUT layout = CblasColMajor;   /* Layout is column major */
CBLAS_UPLO Uplo = CblasUpper;          /* Upper triangle of the matrix */
CBLAS_INDEX N = Ne;                    /* Number of elements in vector X */
float alpha = 1.0;                     /* No scaling, alpha = 1.0 */

/* The vector X */
float complex * X = malloc(Ne * sizeof(* X));

/* Set values of X - for illustration purpose only */
for(int i = 0; i < Ne; i++)
{
  X[i] = CMPLXF(i, i + 1.0);
}

CBLAS_INDEX incX = 1;                  /* Use data from every element */

/* Initialize the array that store correlation matrix */
int size_A = Ne * (Ne + 1) / 2;
float complex * A = malloc(size_A * sizeof(*A));

for(int i = 0; i < size_A; i++)
{
    A[i] = 0.0;
}

cblas_chpr(layout, Uplo, N, alpha, X, incX, A);

/* Print the first value of the result */
float complex print_val = A[0][0];
printf("%+.10f %+.10f", crealf(print_val), cimagf(print_val));

free(X);
free(A);