cblas_dgemm的正确使用

Question

我编写了以下代码来简单地调用 cblas_dgemm 来将两个矩阵相乘。

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <cblas.h>
#define N 2
void fill_matrices(double **first, double **second, double **result) 
{
    srand(time(NULL)); // randomize seed
    for (int i = 0; i < N; i++){
        for (int j = 0; j < N; j++){
            first[i][j] = rand() % 10;
            second[i][j] = rand() % 10;
            result[i][j] = 0;
        }
    }
}
void print(double **first, double **second, double **result) 
{
    printf("First:\n");
    for (int i = 0; i <  N; i++){
        printf("[ ");
        for (int j = 0; j < N; j++){
            printf("%f ", first[i][j]);
        }
        printf("]\n");
    }
    printf("\nSecond:\n");
    for (int i = 0; i <  N; i++){
        printf("[ ");
        for (int j = 0; j < N; j++){
            printf("%f ", second[i][j]);
        }
        printf("]\n");
    }
    printf("\nResult:\n");
    for (int i = 0; i <  N; i++){
        printf("[ ");
        for (int j = 0; j < N; j++){
            printf("%f ", result[i][j]);
        }
        printf("]\n");
    }   
}
int main()
{
    double** first = (double**)malloc(N * sizeof(double*));
    for (int index=0;index<N;++index)
    {
        first[index] = (double*)malloc(N * sizeof(double));
    }
    double** second = (double**)malloc(N * sizeof(double*));
    for (int index=0;index<N;++index)
    {
        second[index] = (double*)malloc(N * sizeof(double));
    }    
    double** result = (double**)malloc(N * sizeof(double*));
    for (int index=0;index<N;++index)
    {
        result[index] = (double*)malloc(N * sizeof(double));
    }
    
    fill_matrices(first, second, result);
 
    cblas_dgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, 
                N,N,N, 1.0, *first, 
                N, *second, N, 0.0, 
                *result, N);

    print(first, second, result);
    return 0;
}

构建命令是

gcc -O3  -c mm_blas.c -o mm_blas.o
g++ -std=c++11 -o  mm_blas mm_blas.o -lblas

但是，程序的输出显示错误的乘法。

First:
[ 1.000000 4.000000 ]
[ 4.000000 4.000000 ]

Second:
[ 8.000000 4.000000 ]
[ 2.000000 6.000000 ]

Result:
[ 8.000000 4.000000 ]
[ 0.000000 0.000000 ]

不仅1*8+4*2写成8，而且第二行全为零。我根据示例使用该语法。我想知道传递给函数的选项是否正确。对此有什么想法吗？

Answer 1

cblas_dgemm 不适用于指向指针的指针，我尝试使用技巧 A[i * N + j] 初始化矩阵，它解决了问题。