CUDA编程之:cudaMemcpy()函数

cudaMemcpy用于在主机(Host)和设备(Device)之间往返的传递数据,用法如下:

主机到设备:cudaMemcpy(d_A,h_A,nBytes,cudaMemcpyHostToDevice)

设备到主机:cudaMemcpy(h_A,d_A,nBytes,cudaMemcpyDeviceToHost)


注意:该函数是同步执行函数,在未完成数据的转移操作之前会锁死并一直占有CPU进程的控制权,所以不用再添加cudaDeviceSynchronize()函数

示例:

复制代码
void VectorReal_copyBuffer(realVecHandle* addr, float hostArr[],long buffer, bool tohost, int copyMode) 
{ 
    realVecHandle * native =addr;
    float * host  = hostArr;

if ( copyMode == 0 ) {
    if (!tohost) {
        cudaMemcpy( native->data,host, native->size, cudaMemcpyHostToDevice) ;
    } else {
        cudaMemcpy( host, native->data, native->size, cudaMemcpyDeviceToHost) ;
    }
}
}
复制代码

 

posted @   手磨咖啡  阅读(1286)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!
点击右上角即可分享
微信分享提示