在CUDA内核中使用cublasStbsv

Question

我试图在我的CUDA内核中用cublasStbsv函数解决方程式。内核代码如下：

__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。有人能帮我吗？

Answer 1

tbsp()期望三角形带状矩阵; tpsv()期望以打包格式存储三角矩阵;并且trsv()期望密集矩阵，并且仅使用上/下部分。

根据你的代码，我认为你需要trsv()。