当前位置:网站首页>cudaMemcpy学习笔记
cudaMemcpy学习笔记
2022-07-31 01:46:00 【AI视觉网奇】
主机是电脑Host
Device是gpu设备。
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) ;
}
}
}
//显存上分配空间
CUDA_SAFE_CALL(cudaMalloc((void**)&Dst_d,sizeof(float3) * totalPNum));
//显存上传输数据
CUDA_SAFE_CALL(cudaMemcpy(Dst_d, Srcdata0_h, sizeof(float3) * numdata0, cudaMemcpyHostToDevice));
CUDA_SAFE_CALL(cudaMemcpy(Dst_d + numpoint0, Srcdata1_d, sizeof(float3) * numdata1, cudaMemcpyDeviceToDevice));
CUDA_SAFE_CALL(cudaMemcpy(Dst_d + numpoint0 + numpoint1, Srcdata2_d, sizeof(float3) * numdata2, cudaMemcpyDeviceToDevice));
数据传输过程中,目的地址在源地址之前(地址可能有偏移计算,数据源不同目的相同时),对应后面参数,devicetohost 或者 hosttodevice.
原文链接:https://blog.csdn.net/xiaoheibaqi/article/details/44001273
边栏推荐
- 【flask入门系列】Flask-SQLAlchemy的使用
- C language _ structure pointer array function voting system
- VSCode Plugin: Nested Comments
- leetcode-952: Calculate max component size by common factor
- MySQL的安装教程(嗷嗷详细,包教包会~)
- 【AcWing 62nd Weekly Game】
- rpm install postgresql12
- 软件测试报告有哪些内容?
- Xiaohei's leetcode journey: 104. The maximum depth of a binary tree
- 《MySQL数据库进阶实战》读后感(SQL 小虚竹)
猜你喜欢
随机推荐
手把手教你配置Jenkins自动化邮件通知
MySql installation and configuration super detailed tutorial and simple method of building database and table
MySQL stored procedure
Crawler text data cleaning
《云原生的本手、妙手和俗手》——2022全国新高考I卷作文
C language _ structure pointer array function voting system
第一学年课程期末考试
最大路径和
Chi-square distribution of digital image steganography
The Meta Metaverse Division lost 2.8 billion in the second quarter, still want to continue to bet?Metaverse development has yet to see a way out
初识C语言 -- 数组
TiKV主要内存结构和OOM排查总结
Meta元宇宙部门第二季度亏损28亿 仍要继续押注?元宇宙发展尚未看到出路
MySQL的分页你还在使劲的limit?
coldfusion8后台计划任务拿shell
What are the project management tools like MS Project
MySql的安装配置超详细教程与简单的建库建表方法
CV-Model [3]: MobileNet v2
Charging effect simulation
ROS Action通信