2012-03-20 53 views
1

我試圖用vecLibs' cblas相乘兩個矩陣:vecLib cblas_sgemm文檔錯誤?

#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

回答

4

您所看到的錯誤似乎對我的眼睛完全正確。

LDA始終是線性內存中陣列A的間距。如果您使用行主要存儲順序,則間距將是列數,而不是行數。所以在這種情況下,LDA應該是3。

+0

該文檔說'lda 矩陣A的第一維的大小;如果你傳遞一個矩陣A [m] [n],那麼這個值應該是m,但這個m是矩陣行的數量。 – Pedro 2012-03-20 18:37:56

+1

文檔說的是正確的 - lda是線性內存*中矩陣A *的第一維。冒號後面的內容適用於列主要訂單的情況。 – talonmies 2012-03-20 18:44:46