4

我正在尝试使用 vecLibs 的 cblas 将两个矩阵相乘:

#include <stdlib.h>
#include <string.h>
#include <stdio.h>

#include <vecLib/cblas.h>

int main (void) {

    float *A = malloc(sizeof(float) *  2 * 3);
    float *B = malloc(sizeof(float) *  3 * 1);
    float *C = malloc(sizeof(float) *  2 * 1);

    cblas_sgemm(CblasRowMajor, 
                CblasNoTrans,
                CblasNoTrans, 
                2,
                1,
                3,
                1.0,
                A, 2,
                B, 3,
                0.0,
                C, 2);

    printf ("[ %f, %f]\n", C[0], C[1]);

    return 0;  
}

根据文档,每个参数似乎都匹配但我收到此错误:

lda 必须 >= MAX(K,1):lda=2 K=3BLAS 错误:传递给 cblas_sgemm 的参数号 9 具有无效值

4

1 回答 1

6

您看到的错误在我看来似乎完全正确。

LDA 总是线性存储器中数组 A 的间距。如果您使用行主要存储顺序,间距将是列数,而不是行数。所以在这种情况下LDA应该是3。

于 2012-03-20T18:31:00.133 回答