我有一个包含主机函数和设备函数的 CUDA 程序执行()。在主机函数中,我分配了一个全局内存output然后它将被传递给设备函数并用于存储设备函数内分配的全局内存的地址。我想访问主机函数中内核分配的内存。以下是代码:
#include <stdio.h>
typedef struct
{
int * p;
int num;
} Structure_A;
__global__ void Execute(Structure_A *output);
int main(){
Structure_A *output;
cudaMalloc((void***)&output,sizeof(Structure_A)*1);
dim3 dimBlockExecute(1,1);
dim3 dimGridExecute(1,1);
Execute<<<dimGridExecute,dimBlockExecute>>>(output);
Structure_A * output_cpu;
int * p_cpu;
cudaError_t err;
output_cpu= (Structure_A*)malloc(sizeof(Structure_A));
err=cudaMemcpy(output_cpu,output,sizeof(Structure_A),cudaMemcpyDeviceToHost);
if( err != cudaSuccess)
{
printf("CUDA error a: %s\n", cudaGetErrorString(err));
exit(-1);
}
p_cpu=(int *)malloc(sizeof(int));
err=cudaMemcpy(p_cpu,output_cpu[0].p,sizeof(int),cudaMemcpyDeviceToHost);
if( err != cudaSuccess)
{
printf("CUDA error b: %s\n", cudaGetErrorString(err));
exit(-1);
}
printf("output=(%d,%d)\n",output_cpu[0].num,p_cpu[0]);
return 0;
}
__global__ void Execute(Structure_A *output){
int thid=threadIdx.x;
output[thid].p= (int*)malloc(thid+1);
output[thid].num=(thid+1);
output[thid].p[0]=5;
}
我可以编译程序。但是当我运行它时,出现错误,表明以下内存复制函数中存在无效参数:
err=cudaMemcpy(p_cpu,output_cpu[0].p,sizeof(int),cudaMemcpyDeviceToHost);
CUDA版本是4.2。
CUDA卡:Tesla C2075
操作系统:x86_64 GNU/Linux
编辑:修改代码并为output_cpu和p_cpu分配适当大小的内存。