CUDA：一维、二维的grid、block的核函数线程分配

1.一维grid，一维block

[cpp] view plain copy

1. int nx = 1 << 14;  
2. int ny = 1 << 14;  
3. int dimx = 32;  
4. dim3 block(dimx, 1);  
5. dim3 grid((nx + block.x - 1) / block.x, 1);  

核函数

[cpp] view plain copy

1. __global__ void kernel_function()  
2. {  
3.     unsigned int ix = threadIdx.x + blockIdx.x * blockDim.x;  
4.   
5.     if (ix < nx )  
6.     {  
7.         do_something();  
8.     }  
9. }  

2.二维grid，一维block

[cpp] view plain copy

1. int nx = 1 << 14;  
2. int ny = 1 << 14;  
3. int dimx = 32;  
4. dim3 block(dimx, 1);  
5. dim3 grid((nx + block.x - 1) / block.x, ny);  

核函数

[cpp] view plain copy

1. __global__ void kernel_function()  
2. {  
3.     unsigned int ix = threadIdx.x + blockIdx.x * blockDim.x;  
4.     unsigned int iy = blockIdx.y;  
5.     unsigned int idx = iy * nx + ix;  
6.   
7.     if (ix < nx && iy < ny)  
8.     {  
9.         do_something();  
10.     }  
11. }  

3.二维grid，二维block

[cpp] view plain copy

1. int nx = 1 << 14;  
2. int ny = 1 << 14;  
3. int dimx = 32;  
4. int dimy = 32;  
5. dim3 block(dimx, dimy);  
6. dim3 grid((nx + block.x - 1) / block.x, (ny + block.y - 1) / block.y);  

核函数

[cpp] view plain copy

1. __global__ void kernel_function()  
2. {  
3.     unsigned int ix = threadIdx.x + blockIdx.x * blockDim.x;  
4.     unsigned int iy = threadIdx.y + blockIdx.y * blockDim.y;  
5.     unsigned int idx = iy * nx + ix;  
6.   
7.     if (ix < nx && iy < ny)  
8.     {  
9.         do_something();  
10.     }  
11. }
    

来源：https://blog.youkuaiyun.com/rong_toa/article/details/78656908