关于定义一个device全局变量如何分配空间的问题

2012-04-06 11:18:32 520 阅读 AI 写同款 · GEO 优化AI 选题数据监测一键分发 ›

lc19890326 墨衍会员 · AI 创作全网分发关注 + 加入社区

MODULE_DEVICE_TABLE函数在你的板子的linux内核有如下文件：/arch/arm/mach-xxxx/board_xxxx.c，该文件中，有struct i2c_board_info __initdata i2c_info，该结构体列表中，有你的这个 wifi driver的i2c信息，这两份信息，通过 wifi_rf_i2c_id 里面的 RDA_WIFI_RF_I2C_DEVNAME 名称来匹配，名称一致，则成功阅读详情

想请教各位，想要在device的代码中，定义一个__device__的全局变量，应该怎么对这个变量分配空间以及将起拷回host端。具体代码如下：

我的程序分两个文件

第一个是.cu，定义了全局变量，kernel函数和三个给全局变量分配空间销毁空间以及拷贝全局变量到host的函数

#include <stdio.h>

#include <stdlib.h>

#include <cuda_runtime.h>

#include <cutil.h>



/************************************************************************/

/* Init CUDA                                                            */

/************************************************************************/

#if __DEVICE_EMULATION__



bool InitCUDA(void){return true;}



#else



#define  SIZE_X 8

#define  SIZE_Y 8

#define  SIZE_Z 8



__device__ float* d_val; //全局变量，存储kernel函数计算后的值





#endif

/************************************************************************/

/* Example                                                              */

/************************************************************************/

__global__ static void HelloCUDA()

{

	for (int z=0;z<SIZE_Z;++z)

	{

		int x = threadIdx.x + blockIdx.x * blockDim.x;

		int y = threadIdx.y + blockIdx.y * blockDim.y;

		while (y<SIZE_Y)

		{

			x = threadIdx.x + blockIdx.x * blockDim.x;

			while(x<SIZE_X)

			{

				d_val[x+SIZE_X*y+SIZE_X*SIZE_Y*z] = 56;/*computePower(d_value,EQUIP_NUM,x,y,z);*/

				x+=blockDim.x*gridDim.x;

			}

			y+=blockDim.y*gridDim.y;

		}

	}

}



extern "C" 

void kernel(dim3 gridSize, dim3 blockSize, float *h_val)

{

	HelloCUDA<<<gridSize,blockSize>>>();

	CUDA_SAFE_CALL( cudaMemcpy(h_val, d_val, sizeof(float)*SIZE_X*SIZE_Y*SIZE_Z, cudaMemcpyDeviceToHost));//最后程序运行，显示此处无效的参数，我觉得应该是kernel函数没有启动，应该是d_val没有分配到空间

}





extern "C"

void InitVal()

{

	CUDA_SAFE_CALL( cudaMalloc((void**) &d_val, sizeof(float) * SIZE_X*SIZE_Y*SIZE_Z)); //给全局变量在分配空间，但好像没分配上

}



extern "C"

void clean()

{

		CUDA_SAFE_CALL(cudaFree(d_val));

}

另外还有一个CPP文件，包含了主函数，和运行以上几个函数，代码也一起贴出来吧

#include <stdio.h>

#include <stdlib.h>

#include <cuda_runtime.h>

#include <cutil.h>





#define  SIZE_X 8

#define  SIZE_Y 8

#define  SIZE_Z 8



extern "C" void kernel(dim3 gridSize, dim3 blockSize, float *h_val);

extern "C" void InitVal();

extern "C" void clean();

 

/************************************************************************/

/* Init CUDA                                                            */

/************************************************************************/

bool InitCUDA(void)

{

	int count = 0;

	int i = 0;



	cudaGetDeviceCount(&count);

	if(count == 0) {

		fprintf(stderr, "There is no device.\n");

		return false;

	}



	for(i = 0; i < count; i++) {

		cudaDeviceProp prop;

		if(cudaGetDeviceProperties(&prop, i) == cudaSuccess) {

			if(prop.major >= 1) {

				break;

			}

		}

	}

	if(i == count) {

		fprintf(stderr, "There is no device supporting CUDA.\n");

		return false;

	}

	cudaSetDevice(i);



	printf("CUDA initialized.\n");

	return true;

}



int iDivUp(int a, int b){

	return (a % b != 0) ? (a / b + 1) : (a / b);

}



/************************************************************************/

/* HelloCUDA                                                            */

/************************************************************************/

int main(int argc, char* argv[])

{



	if(!InitCUDA()) {

		return 0;

	}



	float *h_val;



	h_val = (float*)malloc(SIZE_X*SIZE_Y*SIZE_Z*sizeof(float));



	



	unsigned int timer = 0;

	CUT_SAFE_CALL( cutCreateTimer( &timer));

	CUT_SAFE_CALL( cutStartTimer( timer));



	dim3 blockSize = dim3(8,8);

	dim3 gridSize = dim3(iDivUp(SIZE_X,blockSize.x), iDivUp(SIZE_Y,blockSize.y));



	kernel(gridSize, blockSize,h_val);



	CUT_CHECK_ERROR("Kernel execution failed\n");



	CUDA_SAFE_CALL( cudaDeviceSynchronize() );

	CUT_SAFE_CALL( cutStopTimer( timer));

	printf("Processing time: %f (ms)\n", cutGetTimerValue( timer));

	CUT_SAFE_CALL( cutDeleteTimer( timer));



	



	for (int k= 0;k<SIZE_X;++k)

	{

		for (int j= 0;j<SIZE_Y;++j)

		{

			for (int i= 0;i<SIZE_Z;++i)

				printf("%f\n",h_val[i+j*SIZE_X+k*SIZE_Y*SIZE_X]);

			printf("\n");

		}

		printf("\n");

	}

	printf("-------------->%f", h_val[4]);

	//printf("%f\n",h_val);

	free(h_val);

	clean();

	CUT_EXIT(argc, argv);

	return 0;

}