blas - vecLib cblas_sgemm 文档错误？

Question

我正在尝试使用 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;  
}

根据docs每个参数似乎都匹配，但我收到此错误:

lda must be >= MAX(K,1): lda=2 K=3BLAS error: Parameter number 9 passed to cblas_sgemm had an invalid value

Answer 1

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

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