Использование cublasStbsv в ядре CUDA

Я пытаюсь решить уравнение с помощью функции cublasStbsv в моем ядре CUDA. Код ядра выглядит следующим образом:Использование cublasStbsv в ядре CUDA

__global__ void invokeDeviceCublasSgemm(cublasStatus_t *returnValue, 
             int n, 
             const float *d_alpha, 
             const float *d_A, 
             const float *d_B, 
             const float *d_beta, 
             float *d_C) 
{ 
    cublasHandle_t cnpHandle; 
    cublasStatus_t status = cublasCreate(&cnpHandle); 

    if (status != CUBLAS_STATUS_SUCCESS) 
    { 
     *returnValue = status; 
     return; 
    } 

// /* Perform operation using cublas */ 
// status = 
//  cublasSgemm(cnpHandle, 
//     CUBLAS_OP_N, CUBLAS_OP_N, 
//     n, n, n, 
//     d_alpha, 
//     d_A, n, 
//     d_B, n, 
//     d_beta, 
//     d_C, n); 

    float d_AA[5*5]; 
    float d_BB[5]; 
// float d_X[5]; 

    for(int i=0;i<5;i++) 
    { 
     for(int j=0;j<5;j++) 
     { 
      if(i==j) 
      { 
       d_AA[i*5+j] = i; 
      }else 
      { 
       d_AA[i*5+j] = 0; 
      } 

     } 
     d_BB[i] = i*i; 
    } 

    status = cublasStbsv(cnpHandle, 
         CUBLAS_FILL_MODE_UPPER, 
         CUBLAS_OP_N, 
         CUBLAS_DIAG_NON_UNIT, 
         n,n, 
         d_AA, 
         5, 
         d_BB, 
         1); 

    for(int i=0;i<5;i++) 
    { 
      printf("B i %d %f \n",i,d_BB[i]); 
    } 

    cublasDestroy(cnpHandle); 

    *returnValue = status; 
}

Я не понимаю, почему я получаю следующее сообщение об ошибке:

Starting /home/xavier/Bureau/Developpement/Cuda/build-Cuda_CUBLAS-Qt_5_6_0_gcc_64-Release/Cuda_CUBLAS...

simpleDevLibCUBLAS test running...
GPU Device 0: "GeForce GTX 750 Ti" with compute capability 5.0

Host and device APIs will be tested.
** On entry to SBSV parameter number 7 had an illegal value
B i 0 0.000000
B i 1 1.000000
B i 2 4.000000
B i 3 9.000000
B i 4 16.000000
!!!! CUBLAS Device API call failed with code 7

Я не понимаю, какую функцию я должен использовать, чтобы решить мою линейную уравнение - cublasStpsv или cublasStrs. Кто-нибудь может мне помочь?

источник

2016-06-09 user3262532

tbsp() ожидает треугольную матрицу; tpsv() ожидает треугольную матрицу, хранящуюся в упакованном формате; и trsv() ожидает плотную матрицу, и будет использоваться только верхняя/нижняя часть.

В соответствии с вашим кодом, я думаю, вам нужно trsv().

источник

2016-06-09 12:45:22 kangshiyin

Какую функцию следует использовать для решения линейного уравнения, например A = m * n B = m * 1 и X = m * 1. Думаю, в trsv(), A должен был быть квадратным? – user3262532

@ user3262532 Эти 3 API математически идентичны друг другу - все они требуют квадратной матрицы, только схемы хранения данных A различны. Если m> n, это линейная задача с наименьшим квадратом - это математическая проблема, а не проблема программирования. – kangshiyin

Как вы можете это сделать, когда у вас есть больше уравнения, которое неизвестно в вашем уравнении? Существует ли функция, которая решает A X = 0 с A не квадратом? – user3262532

Использование cublasStbsv в ядре CUDA

ответ

Смежные вопросы