cuda段错误

Question

我正在使用 C 中的 cuda 进行编程。我遇到了以下代码行的段错误:

int width = 0;
int height = 0;

// load input image
unsigned char * image_input = loadFile(&width, &height, "test.bmp");

// allocate memory for output
unsigned char * image_output = (unsigned char *) malloc(width*height*sizeof(unsigned char));

// set the size of the input and out array 2D
int size = width*height*sizeof(int);

// Allocate space on the GPU for input and output
char* GPU_input = 0;
char* GPU_output = 0;   

cudaMalloc(&GPU_input, size);       
cudaMalloc(&GPU_output, size);

// Copy the input data to the GPU (host to device)
cudaMemcpy(GPU_input, image_input, size, cudaMemcpyHostToDevice); //segmentation fault here

有什么想法吗？

提前致谢。

Answer 1

• 大小是 sizeof(int) *H*W。 (以字节为单位)
• Image_input 是 H*W。 (以字节为单位)

您正在处理超出其大小的 image_input。