首页 > 代码库 > CUDA __shared__ thread、block、grid之间的一维关系 （例子chapter5 dot点积（GPU高性能编程））

CUDA __shared__ thread、block、grid之间的一维关系 （例子chapter5 dot点积（GPU高性能编程））

2024-07-09 00:29:41   220人阅读

chapter5里重要的例子是dot，来解释一个block内多个thread的共享内存和同步。

__shared__共享内存：“对于在GPU上启动的每个线程块，cuda c编译器都将创建该变量的一个副本。线程块中的每个线程都共享这块内存，并和其他线程块无关，这使一个线程块中多个线程能够在计算上进行通信和协作”

__syncthreads()：确保线程块中的每个线程都执行完__syncthreads()前面的语句后，在往下执行。

例子是Grid->一维Block->一维Thread：

 

通过实例代码来分析：

#include <iostream>#include "cuda_runtime.h"#include "device_launch_parameters.h"#include "book.h"#include "cpu_bitmap.h"//1.minFun  对于每个Grid而言，选取block数比较少的，这样可以减少每个grid中最后一个block内thread的冗余。int imin(int a,int b){	if(a>b)	{		return b;	}else{		return a;	}}const int N = 33*1;//整体线程的数量，即向量（a）*(b)中每个向量数组的大小const int ThreadNumsPerBlock =256;//每个block中thread的数量const int BlockNumsPerGrid = imin(32,(N+ThreadNumsPerBlock-1)/ThreadNumsPerBlock);//每个grid中block的数量，__global__ void dot(float* a,float* b,float* c)//输入两个向量：a，b ；结果c大小为一个grid内block的大小，grid内每个block的位置对应数组c每个下标对应的数据，每个数据为多个grid相同的block下标下的thread的和，如图-1。{   int	everyThreadIndex  =  threadIdx.x+blockIdx.x*blockDim.x;//获取一个grid中thread的index   int  cacheIndex = threadIdx.x;//获取一个block中thread的index   __shared__ float cache[ThreadNumsPerBlock];   float temp = 0;   while(everyThreadIndex<N)//多个grid的跳跃index，   {	   temp += a[everyThreadIndex]*b[everyThreadIndex];	   everyThreadIndex += blockDim.x*gridDim.x;//   }   cache[cacheIndex] = temp;//得到多个grid中 一个block里相应的一个thread的乘积的和   __syncthreads();   int i= blockDim.x/2;   while(i!=0)   //一直纳闷这个怎么在多个线程中执行啊？应该是：一个thread执行到这里后，__syncthreads()之前一个语句等啊等，等到所有一个block里的threads全都执行一遍后，才往下进行   {	   if(cacheIndex<i)	   {		   cache[cacheIndex]= cache[cacheIndex]+cache[cacheIndex+i];//while里的这句是在block里不同的thread里执行的	   }	   __syncthreads();	   i=i/2;//当每个thread都执行后，才能进行长度折半   }   if(0==cacheIndex)   {     c[blockIdx.x] = cache[0];   }}int main(void)//{	float  *host_a,*host_b,*host_c;	float  *device_a,*device_b,*device_c;	host_a = (float*)malloc(N*sizeof(float));	host_b = (float*)malloc(N*sizeof(float));	host_c = (float*)malloc(BlockNumsPerGrid*sizeof(float));	  for (int i=0; i<N; i++) {        host_a[i] = i;        host_b[i] = i*2;    }	//gpu上分配内存	cudaMalloc((void**)&device_a,N*sizeof(float));	cudaMalloc((void**)&device_b,N*sizeof(float));	cudaMalloc((void**)&device_c,BlockNumsPerGrid*sizeof(float));	cudaMemcpy(device_a,host_a,N*sizeof(float),cudaMemcpyHostToDevice);	cudaMemcpy(device_b,host_b,N*sizeof(float),cudaMemcpyHostToDevice);	dot<<<BlockNumsPerGrid,ThreadNumsPerBlock>>>(device_a,device_b,device_c);	cudaMemcpy(host_c,device_c,BlockNumsPerGrid*sizeof(float),cudaMemcpyDeviceToHost);    	float c=0;	for(int i=0;i<BlockNumsPerGrid;i++)	{		c = c + host_c[i];	}	  #define sum_squares(x)  (x*(x+1)*(2*x+1)/6)	std::cout<<"结果1："<<c<<"结果2："<<2 * sum_squares( (float)(N - 1) )<<std::endl;	cudaFree(device_a);	cudaFree(device_b);	cudaFree(device_c);	free(host_a);	free(host_b);	free(host_c);	return 0;}

书中最后讲解了错误的修改方式：

while(i!=0)   {	   if(cacheIndex<i)	   {		   cache[cacheIndex]= cache[cacheIndex]+cache[cacheIndex+i];//while里的这句是在block里不同的thread里执行的
                   __syncthreads();
            }
         i=i/2;//当每个thread都执行后，才能进行长度折半 
   }

原则是“线程执行相同的指令，不同的数据运算”

__syncthreads()中，cuda架构要确保一个block中每个thread都执行__syncthreads()，否则没有任何thread执行__syncthreads之后的操作。

当一些thread执行一条指令，而其他线程不需要时，这种情况称作线程发散（Thread Divergence）.这种情况下要注意__syncthreads的位置。

 

小结：

1. dim3 a(x,y); dim3 b(x,y);         <<<a,b>>>  a：Grid中启动block的数量 b:Block中启动thread数量

2.注意线程块的向后取整（取上限）；  还有通过  if(threadIndex<N)来确保thread的范围

3.__shared__和__syncthreads()的正确用法，并注意线程发散情况

4.使用线程与使用线程块相比有什么优势？ ：一：解决线程块数量的限制  二：进行部分的数据共享和通讯

5.一维：threadIdx.x: block中的thread索引   blockIdx.x :Grid中block索引   blockDim.x：一个Block中thread数量; gridDim.x:一个Grid中Block数量  


联系
我们