给cuda核函数传递二维数组的一种方法

#include <stdio.h>

/**
 * 需求：需要把若干个一维数组传给核函数
 * 实现方法：在gpu生成一个一维的指针数组，每个元素指向一个普通一维数组。
 * 把该指针数组的地址传递给核函数。
 * 其实该指针数组充当二维数组的角色。
 */

__global__ void
testKernel(float ** pointerArray)
{
	printf("(%d,%d):%f\n",threadIdx.y,threadIdx.x,pointerArray[threadIdx.y][threadIdx.x]);
}


int main()
{

	int rows = 2;
	int cols = 2;


	float** host_2d = new float*[rows];
	float** dev_2d;
	for (int k = 0; k < rows; k++)
	{
		float* host_1d = new float[cols];
		//---------下面两个是测试数据，看看核函数printf输出的是否与这个一致------------------------------
		host_1d[0] = 0;
		host_1d[1] = 1;

		float* dev_1d ;
		cudaMalloc((void**)&dev_1d,sizeof(float)*cols);//该指针指向的是一个float数组
		cudaMemcpy(dev_1d,host_1d,sizeof(float)*cols,cudaMemcpyHostToDevice);

		host_2d[k] = dev_1d;
	}
	cudaMalloc((void**)&dev_2d,sizeof(float*)*rows);
	cudaMemcpy(dev_2d,host_2d,sizeof(float*)*rows,cudaMemcpyHostToDevice);

	dim3 threads(rows,cols);
	testKernel<<<1,threads>>>(dev_2d);

	cudaDeviceSynchronize();

	//记得释放内存啊，养成好习惯。这里省去这部分代码




return 0;
}

输出结果：

(0,0):0.000000
(0,1):1.000000
(1,0):0.000000
(1,1):1.000000

与猜想一致