轻云流风 2014-04-20 03:54
浏览 2279

cuda向设备端传递多个一维数组,并在核函数修改数组元素值

现在有多个一维数组,我想将他们拷贝到设备端修改他们的值,然后拷贝回host端,代码如下:
#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include "helper_cuda.h"
#include
#include
using namespace std;
#define N 10
global void gpuKernel(int ptr)
{

int tid=threadIdx.x;
int bid=blockIdx.x;
if (tid<N &&bid<N)
{

ptr[bid][tid]=N;

}
}
int main()
{
int **hptr,
hostArray;
int dptr,devArray;
hptr=new int [N];
cudaMalloc(&dptr,N*sizeof(int
));

hostArray=new int *[N];
devArray=new int *[N];
for (int i=0;i<N;i++)
{
hostArray[i]=new int[N];
for (int j=0;j<N;j++)
{
hostArray[i][j]=1;
}
}
for (int i=0;i<N;i++)
{
cudaMemcpy(devArray[i],hostArray[i],N*sizeof(int),cudaMemcpyHostToDevice);
hptr[i]=devArray[i];//保存每个数组的头指针
}

cudaMemcpy(dptr,hptr,N*sizeof(int*),cudaMemcpyHostToDevice);    
dim3 dimblock=N;
dim3 dimgrid =N;
gpuKernel<<<dimgrid,dimblock>>>(dptr);
for (int i=0;i<N;i++)
{       cudaMemcpy(hostArray[i],devArray[i],N*sizeof(int),cudaMemcpyDeviceToHost);      
}
for (int i=0;i<N;i++)
{
    for (int j=0;j<N;j++)
    {
        cout<<hostArray[i][j]<<",";
    }
    cout<<endl;

}
//free memory
for (int i=0;i<N;i++)
{
    delete []hostArray[i];
    cudaFree(devArray[i]);
}
cudaFree(dptr);
/*cudaFree(devArray);*/
delete[]hptr;
delete []hostArray;
delete[]devArray;   

return 0;

}
最终结果是元素值都为修改,纠结这个问题几天了,并且仔细拜读bendanban大神的博客http://blog.csdn.net/bendanban/article/details/7669624,可能自己理解有误,望各位大神指点

  • 写回答

0条回答 默认 最新

    报告相同问题?

    悬赏问题

    • ¥20 有关区间dp的问题求解
    • ¥15 多电路系统共用电源的串扰问题
    • ¥15 slam rangenet++配置
    • ¥15 有没有研究水声通信方面的帮我改俩matlab代码
    • ¥15 对于相关问题的求解与代码
    • ¥15 ubuntu子系统密码忘记
    • ¥15 信号傅里叶变换在matlab上遇到的小问题请求帮助
    • ¥15 保护模式-系统加载-段寄存器
    • ¥15 电脑桌面设定一个区域禁止鼠标操作
    • ¥15 求NPF226060磁芯的详细资料