C++的數據結構和CUDA

Question

我有其可以是

struct type1{ double a,b,c;} 

或結構也可以是

struct type2{ double a,b,c,d,e;} 

在CUDA代碼我的主機功能我有成纔像

void compute(){ 
     // some code 
     // data on devices (up to 10) 
     type *xxx[10]; // this is where i want either type1 or type2 structures 
         // the "type" is not known at compile time but i want to 
         // determine at runtime based on some input variable. this 
         // part is not real code rather this is what i want to achive. 

     int DevUsed; // some code to give value to int DevUsed 

     for(int idev=0;idev<DevUsed;idev++){ 

      // set cuda device 

      if (cudaMalloc(&xxx[iDev], sizeof(type)) != cudaSuccess) 
      // print error message; 
      cudaMemcpy(xxx[iDev], pIF1, sizeof(type), cudaMemcpyHostToDevice); 

      function2<<<grid, block>>>(xxx[iDev]); // where function2 is the kernel 
     } 
    } 

我的問題是什麼是使用類型爲「type * xxx [10];」的泛型代碼在type1和type2數據結構之間進行選擇的方法。

Answer 1

C++模板是針對這種情況而設計的。

template <class T> 
void compute(){ 
    // some code 
    // data on devices (up to 10) 
    T xxx[10]; // this is where i want either type1 or type2 structures 
        // the "type" is not known at compile time but i want to 
        // determine at runtime based on some input variable. this 
        // part is not real code rather this is what i want to achive. 

    int DevUsed; // some code to give value to int DevUsed 

    for(int idev=0;idev<DevUsed;idev++){ 

     // set cuda device 

     if (cudaMalloc(&xxx[iDev], sizeof(T)) != cudaSuccess) 
     // print error message; 
     cudaMemcpy(xxx[iDev], pIF1, sizeof(T), cudaMemcpyHostToDevice); 

     function2<<<grid, block>>>(xxx[iDev]); // where function2 is the kernel 
    } 
} 

請注意，您還需要像

template <class T> 
__global__ void function2(T x) 
{ 
    //.... 
}