使用GEMM的cublas AA‘

Question

我试图找到计算AA的乘法，其中A是MxN，我试着这样做：

    double al =1.0f; // al =1
    double bet =0.0f; // bet =1
stat=cublasDgemm(handle,CUBLAS_OP_N,CUBLAS_OP_T,M,M,N,&al,d_A,M,d_Acopy,M,&bet,d_temp,M);

但是它给出了分割误差I found this question but I couldn't follow

你能帮我做AA吗，知道我用双指针手动初始化了矩阵

double ** m = (double**)malloc(rows * sizeof(double*));

另外，我将这些值保存为

int i,j;
//initalize A
for (i=0;i<cols;i++)
    {
     for (j=0;j<rows;j++)
       {
        a[i][j]=A_example[i*rows+j];
        //a[i][j]=my_round(a[i][j]*10000.0)/10000.0;
        }
    }

Answer 1

我只是重新定义了我的矩阵，就像他们在示例中定义的那样。我认为问题出在他们使用的数据布局上。我无法将我的C矩阵转换为cuBlas格式。

不管怎么说:他们这样定义它：

double *m=( double *) malloc (rows*cols* sizeof ( double ));

对于数据初始化: int i；int j；

for (i=0;i<k;i ++){
    for (j=0;j<n;j ++){
        printf (" %5.0f",a[ IDX2C(i,j,k )]);
        }

其中，IDX2C是定义为：

# define IDX2C(i,j,ld) (((j)*( ld ))+( i ))

最后我找到了AA‘，如下所示：

double al =1.0f; // al =1
double bet =0.0f; // bet =1
stat=cublasDgemm(handle,CUBLAS_OP_N,CUBLAS_OP_T,M,M,N,&al,d_A,N,d_A,M,&bet,d_temp,M);

其中A是MxN矩阵。