gpu - 组织 CUDA 内核

Question

我正在编写一个非常非常长的 CUDA 内核，它对人类的可读性来说非常糟糕。有什么方法可以用内核外部的功能组织 CUDA 内核吗？示例:

__global__ void CUDA_Kernel(int* a, int* b){
     //calling function 1
     //calling function 2
     //calculation function
         .......
}

Answer 1

如果函数是使用 __device__ 关键字定义的，则可以从内核内部调用该函数。

例如:

__device__ int test_fun(int val)
{
   return 2*val + 3;
}


__global__ void kern_test( int * data) 
{
   int aOffset = blockDim.x * blockIdx.x + threadIdx.x;
   data[offset] = test_fun(data[offset]);
}