Cblas_sgemm производит неправильные результаты

Question

У меня проблема с использованием функции sgemm cblas.

Вот код:

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

#define MATRIX_DIM 5 

int main(){ 

    float *a_mat = calloc(MATRIX_DIM*MATRIX_DIM, sizeof(float)); 
    float *b_mat = calloc(MATRIX_DIM, sizeof(float)); 
    float *c_mat = calloc(MATRIX_DIM, sizeof(float)); 
    int i,j; 

    for(i=0; i<MATRIX_DIM*MATRIX_DIM; i++) { 
     a_mat[i] = 1.0f; 
     b_mat[i] = 1.0f; 
     c_mat[i] = 0.0f; 
    } 

    cblas_sgemm(CblasRowMajor, CblasNoTrans, 
       CblasNoTrans, MATRIX_DIM, MATRIX_DIM, 
       MATRIX_DIM, 1.0, a_mat, 
       MATRIX_DIM, b_mat, MATRIX_DIM, 
       1.0, c_mat, MATRIX_DIM); 

    //RESULT PRINTING 
    printf("Printing A MATRIX: \n"); 
    for(i=0; i<MATRIX_DIM; i++) { 
     for(j=0; j<MATRIX_DIM; j++){ 
       printf("%0.1f ", a_mat[i*MATRIX_DIM+j]); 
     } 
     printf("\n"); 
    } 
    printf("Printing B MATRIX: \n"); 
    for(i=0; i<MATRIX_DIM; i++) { 
      for(j=0; j<MATRIX_DIM; j++){ 
       printf("%0.1f ", b_mat[i*MATRIX_DIM+j]); 
     } 
     printf("\n"); 
    } 

    printf("\nPrinting the Results: \n"); 
    for(i=0; i<MATRIX_DIM;i++){ 
     for(j=0; j<MATRIX_DIM; j++){ 
       printf("%0.1f ", c_mat[i*MATRIX_DIM+j]); 
     } 
     printf("\n"); 
    } 

    free(a_mat); 
    free(b_mat); 
    free(c_mat); 

    return 0; 
} 

Я довольно некоторые из аргументов, я ставлю в ошибаемся, но я действительно не знаю. Результаты должны быть матрицей 5x5, заполненной 5.0. Вместо этого программа отвечает следующим образом:

6.0 6.0 6.0 16.0 86.0 
6.0 6.0 6.0 16.0 86.0 
16.0 36.0 6.0 46.0 86.0 
16.0 36.0 5.0 45.0 85.0 
20.0 80.0 5.0 45.0 85.0 

Я знаю rowmajor порядка или аргументы транспонирования могут быть ошибочными, и я понять те позже, но в данном конкретном умножении ответ должен быть 5,0 или иначе.

Answer 1

Благодаря @AndrasDeak в комментариях все, что мне было нужно, это выделить больше пространства на две матрицы, которые я ранее забыл.

Так в основном меняется:

float *b_mat = calloc(MATRIX_DIM, sizeof(float)); 
float *c_mat = calloc(MATRIX_DIM, sizeof(float)); 

To:

float *b_mat = calloc(MATRIX_DIM*MATRIX_DIM, sizeof(float)); 
float *c_mat = calloc(MATRIX_DIM*MATRIX_DIM, sizeof(float)); 

С тех должны быть 2-мерные матрицы, а не векторы.