使用CUDA C添加两个向量,启动内核

时间:2012-12-09 16:57:23

标签: cuda

我在异构并行编程中做了一些功课。代码由教学人员编写,我们的职责是填充//@@标记的区域。代码应该使用CUDA C添加两个向量。我已经尝试了下面的解决方案,虽然程序执行没有错误,但反馈表明代码的输出与预期结果不匹配。在我添加了我认为需要的代码后,这是代码:

// MP 1
#include    <wb.h>

__global__ void vecAdd(float* in1, float* in2, float* out, int len) {
//@@ Insert code to implement vector addition here
int i = threadIdx.x + blockDim.x * blockIdx.x;
if (i < len ) out[i] = in1[i] + in2[i]; 
}



int main(int argc, char ** argv) {
wbArg_t args;
int inputLength;
float * hostInput1;
float * hostInput2;
float * hostOutput;
float * deviceInput1;
float * deviceInput2;
float * deviceOutput;
//@@ i added ######
int size = inputLength*sizeof(float);
//@@ ########
args = wbArg_read(argc, argv);

wbTime_start(Generic, "Importing data and creating memory on host");
hostInput1 = (float *) wbImport(wbArg_getInputFile(args, 0), &inputLength);
hostInput2 = (float *) wbImport(wbArg_getInputFile(args, 1), &inputLength);
hostOutput = (float *) malloc(inputLength * sizeof(float));

wbTime_stop(Generic, "Importing data and creating memory on host");

wbLog(TRACE, "The input length is ", inputLength);

wbTime_start(GPU, "Allocating GPU memory.");
//@@ Allocate GPU memory here

cudaMalloc((void**)&deviceInput1 , size);
cudaMalloc((void**)&deviceInput2 , size);
cudaMalloc((void**)&deviceOutput , size);
wbTime_stop(GPU, "Allocating GPU memory.");

wbTime_start(GPU, "Copying input memory to the GPU.");
//@@ Copy memory to the GPU here
cudaMemcpy(deviceInput1, hostInput1, size, cudaMemcpyHostToDevice);
cudaMemcpy(deviceInput2, hostInput2, size, cudaMemcpyHostToDevice);
wbTime_stop(GPU, "Copying input memory to the GPU.");

//@@ Initialize the grid and block dimensions here  
dim3 DimGrid((inputLength -1)/256 +1 , 1 , 1);
dim3 DimBlock(256 , 1, 1); 

wbTime_start(Compute, "Performing CUDA computation");
//@@ Launch the GPU Kernel here     
vecAdd<<<DimGrid , DimBlock>>>(deviceInput1 , deviceInput2 , deviceOutput , inputLength); 
cudaThreadSynchronize();
wbTime_stop(Compute, "Performing CUDA computation");

wbTime_start(Copy, "Copying output memory to the CPU");
//@@ Copy the GPU memory back to the CPU here
cudaMemcpy(hostOutput, deviceOutput, size , cudaMemcpyDeviceToHost);
wbTime_stop(Copy, "Copying output memory to the CPU");

wbTime_start(GPU, "Freeing GPU Memory");
//@@ Free the GPU memory here
free(deviceInput1);
free(deviceInput2);
free(deviceOutput);

wbTime_stop(GPU, "Freeing GPU Memory");

wbSolution(args, hostOutput, inputLength);

free(hostInput1);
free(hostInput2);
free(hostOutput);

return 0;
}  

2 个答案:

答案 0 :(得分:2)

将代码移到inputLength变量已获得正确值的位置。改变这个:

//@@ i added ######
int size = inputLength*sizeof(float);
//@@ ########
args = wbArg_read(argc, argv);

wbTime_start(Generic, "Importing data and creating memory on host");
hostInput1 = (float *) wbImport(wbArg_getInputFile(args, 0), &inputLength);
hostInput2 = (float *) wbImport(wbArg_getInputFile(args, 1), &inputLength);
hostOutput = (float *) malloc(inputLength * sizeof(float));

到此:

args = wbArg_read(argc, argv);

wbTime_start(Generic, "Importing data and creating memory on host");
hostInput1 = (float *) wbImport(wbArg_getInputFile(args, 0), &inputLength);
hostInput2 = (float *) wbImport(wbArg_getInputFile(args, 1), &inputLength);
hostOutput = (float *) malloc(inputLength * sizeof(float));

//@@ i added ######
int size = inputLength*sizeof(float);
//@@ ########

此外,请在评论中做出talonmies的建议。

答案 1 :(得分:1)

非常感谢talonmies和ahmad他们都帮助我找到了对我有用的正确答案,完整的答案(对谁来说很有趣)如下:

// MP 1
#include    <wb.h>

__global__ void vecAdd(float* in1, float* in2, float* out, int len) {
int i = threadIdx.x + blockDim.x * blockIdx.x;
if (i < len ) out[i] = in1[i] + in2[i];
 }



int main(int argc, char ** argv) {
wbArg_t args;
int inputLength;
float * hostInput1;
float * hostInput2;
float * hostOutput;
float * deviceInput1;
float * deviceInput2;
float * deviceOutput;


args = wbArg_read(argc, argv);

wbTime_start(Generic, "Importing data and creating memory on host");
hostInput1 = (float *) wbImport(wbArg_getInputFile(args, 0), &inputLength);
hostInput2 = (float *) wbImport(wbArg_getInputFile(args, 1), &inputLength);
hostOutput = (float *) malloc(inputLength * sizeof(float));
int size = inputLength*sizeof(float);

wbTime_stop(Generic, "Importing data and creating memory on host");

wbLog(TRACE, "The input length is ", inputLength);

wbTime_start(GPU, "Allocating GPU memory.");


cudaMalloc((void**)&deviceInput1 , size);
cudaMalloc((void**)&deviceInput2 , size);
cudaMalloc((void**)&deviceOutput , size);
wbTime_stop(GPU, "Allocating GPU memory.");

wbTime_start(GPU, "Copying input memory to the GPU.");

cudaMemcpy(deviceInput1, hostInput1, size, cudaMemcpyHostToDevice);
cudaMemcpy(deviceInput2, hostInput2, size, cudaMemcpyHostToDevice);
wbTime_stop(GPU, "Copying input memory to the GPU.");


dim3 DimGrid((inputLength -1)/256 +1 , 1 , 1);
dim3 DimBlock(256 , 1, 1); 

wbTime_start(Compute, "Performing CUDA computation");
//@@ Launch the GPU Kernel 

vecAdd<<<DimGrid , DimBlock>>>(deviceInput1 , deviceInput2 , deviceOutput ,  inputLength);

cudaThreadSynchronize();
wbTime_stop(Compute, "Performing CUDA computation");

wbTime_start(Copy, "Copying output memory to the CPU");
//@@ Copy the GPU memory back to the CPU 
cudaMemcpy(hostOutput, deviceOutput, size , cudaMemcpyDeviceToHost);
wbTime_stop(Copy, "Copying output memory to the CPU");

wbTime_start(GPU, "Freeing GPU Memory");
//@@ Free the GPU memory 
cudaFree(deviceInput1);
cudaFree(deviceInput2);
cudaFree(deviceOutput);

wbTime_stop(GPU, "Freeing GPU Memory");

wbSolution(args, hostOutput, inputLength);

free(hostInput1);
free(hostInput2);
free(hostOutput);

return 0;
}