cublas умножения матриц, как ожидалось

Question

Я пытаюсь заменить мой GPU блок умножения матриц с cublas, но я не получаю то, что я ожидал на тестовом 2х2 случае:

#include "cuda_runtime.h" 
#include "cublas_v2.h" 
#include "stdio.h" 
#include "omp.h" 


int main(int argc, char **argv) { 

    const int SZ = 2; 
    const size_t MB = SZ*SZ*sizeof(float); 

    cudaSetDevice(0); 

    float *m1, *m2, *m3; 
    float *m1_, *m2_, *m3_; 

    unsigned int i, j; 

    m1 = (float *)malloc(MB); 
    m2 = (float *)malloc(MB); 
    m3 = (float *)malloc(MB); 

    cudaMalloc((float **)&m1_, MB); 
    cudaMalloc((float **)&m2_, MB); 
    cudaMalloc((float **)&m3_, MB); 

    for (i=0; i<SZ*SZ; i++) { 
    j = (int) (i==1); 
    m1[i] = j; 
    j = (int) (i==3); 
    m3[i] = j; 
    printf("m1[%d]=%f m3[%d]=%f\n",i,m1[i],i,m3[i]); 
    } 

    cublasHandle_t handle; 
    cublasCreate(&handle); 

    cublasSetMatrix(SZ,SZ,MB,m1,SZ,m1_,SZ); 
    cublasSetMatrix(SZ,SZ,MB,m3,SZ,m3_,SZ); 

    float al = 1.0f; 
    float bt = 0.0f; 

    cublasSgemm(handle,CUBLAS_OP_N,CUBLAS_OP_N,SZ,SZ,SZ,&al,m3_,SZ,m1_,SZ,&bt,m2_,SZ); 
    printf("\n%s\n\n", cudaGetErrorString(cudaDeviceSynchronize())); 
    cublasGetMatrix(SZ,SZ,MB,m2_,SZ,m2,SZ); 

    for (i=0; i<SZ*SZ; i++) 
    printf("m2[%d]=%f\n",i,m2[i]); 

    free(m1); 
    free(m2); 
    free(m3); 

    cublasDestroy(handle); 

    cudaFree(m1_); 
    cudaFree(m2_); 
    cudaFree(m3_); 


    cudaDeviceReset(); 

    return 0; 
} 

Так что я ожидал для m2 вернуть следующая матрица:

[0 1 
0 0] 

в результате умножения

[0 1 
0 0] 

и

[0 0 
0 1] 

Я используем обращенный заказ на m1 и m3 так, что должно дать мне правильный выход для m2, извлекаемые при условии, что cublas {Set/Get} Матрица работает в столбцах. Но вот вывод кода:

m1[0]=0.000000 m3[0]=0.000000 
m1[1]=1.000000 m3[1]=0.000000 
m1[2]=0.000000 m3[2]=0.000000 
m1[3]=0.000000 m3[3]=1.000000 

no error 

m2[0]=0.000000 
m2[1]=0.000000 
m2[2]=0.000000 
m2[3]=0.000000 

Я не знаю, что я здесь делаю неправильно; Я был бы очень признателен за ваш вклад.

Answer 1

Прототип cublasGetMatrix является:

cublasStatus_t cublasGetMatrix(int rows, int cols, int elemSize, 
         const void *A, int lda, void *B, int ldb); 

elemSize должен быть размер одного элемента матрицы (т.е. sizeof(float)). Это то же самое для cublasSetMatrix:

cublasStatus_t cublasSetMatrix(int rows, int cols, int elemSize, 
         const void *A, int lda, void *B, int ldb)