GPU并行——循环展开_for循环怎么放到gpu并行

    利用GPU进行多线程并行计算，将for循环程序块中单次循环的计算工作分配到GPU的单个线程中去，由此多层嵌套循环可被展开。下面举例介绍如何快速将线程标号对应到相应循环标号上。

    现有如下的嵌套循环：

    要对整个循环进行展开需要用到nelx  * nely* nelz个线程，为了适应硬件对线程束大小的规定，线程块设为32*32大小，具体线程分配如下：

#define BLOCK_WIDTH 1024
 
//分配线程
	int bkX1 = nelx  * nely* nelz /BLOCK_WIDTH;  // X方向 
	if (nelx * nely * nelz % BLOCK_WIDTH) bkX1++;
 
	dim3 grid(bkX1); // 默认y,z维度为1
	dim3 block(32,32);// 默认z维度为1

    接下来实现线程标号对应至循环标号

    根据第一幅图中的e1变量的赋值语句可知，e1变量为循环次数的总标号，将线程标号idx和e1对应起来理解，根据e1总标号反推出i1、j1、k1循环标号，也就是将线程标号对应到循环标号，此问题较容易：

//线程标号
  int idx = threadIdx.x * blockDim.y + threadIdx.y  + blockDim.x * blockDim.y * blockIdx.x;
//余数	
  int remainder;
//最外层循环k1	
  int k1=idx/ (nelx * nely);
//取余
  remainder =idx% (nelx * nely);
//内层循环j1	
  int j1 = remainder / nelx;
  remainder = remainder % (nelx);
//内层循环i1	
  int i1 = remainder;

    将idx理解为式(k1)*nelx*nely+(j1)*nelx+i1，由于nelx、nely、nelz给定，idx首先对最外层循环进行除整以及求余操作，再由remainder对内层循环进行相同操作即可依次获得k1、j1、i1，由此，一个三层循环被展开。