【CUDA学习】结构体指针复制(好)

内核函数中要用data结构作用参数

1 typedef struct
2 {
3     int* value;
4     int* num;
5 } data;

复制代码

 1 //host端
 2 
 3 data* h_input;
 4 h_input=(data*)malloc(sizeof(data));
 5 h_input->value=(int*)malloc(sizeof(int)*N);
 6 h_input->num=(int*)malloc(sizeof(int)*N);
 7 
 8 data tmp;    //用于过渡
 9 cudaMalloc((void**)& tmp->value,sizeof(int)*N);
10 cudaMalloc((void**)&tmp->num,sizeof(int)*N);

复制代码

复制代码

1 //device端
2 
3 data* d_input;
4 cudaMalloc((void**)&d_input,sizeof(data));
5 
6 //将host端数据复制端device端
7 cudaMemcpy(tmp->value,h_input->value,sizeof(int)*N,cudaMemcpyHostToDevice);
8 cudaMemcpy(tmp->num,h_input->num,sizeof(int)*N,cudaMemcpyHostToDevice);
9 cudaMemcpy(d_input,&tmp,sizeof(data),cudaMemcpyHostToDevice);

复制代码

结构体复制的示意图如下所示:

转:https://www.cnblogs.com/dwdxdy/p/3215103.html

猜你喜欢

转载自blog.csdn.net/eric_e/article/details/88747298