Документация vecLib cblas_sgemm неправильная? - PullRequest
Новая
       27

Документация vecLib cblas_sgemm неправильная?

2 голосов
/ 20 марта 2012

Я пытаюсь умножить две матрицы, используя cblas vecLibs: 

#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 должно быть> = MAX (K, 1): lda = 2 K = ошибка 3BLAS: параметр номер 9, переданный в cblas_sgemm, имел недопустимое значение

1 Ответ

4 голосов
/ 20 марта 2012

Ошибка, которую вы видите, кажется мне совершенно правильной.

LDA - это всегда шаг массива A в линейной памяти.Если вы используете порядок строк порядок хранения, то шагом будет количество столбцов, а не количество строк.Таким образом, LDA должно быть 3 в этом случае.

...