CUDA GPU之间通过P2P通信进行数据的拷贝

最新推荐文章于 2023-03-03 10:24:44 发布

Coder802

最新推荐文章于 2023-03-03 10:24:44 发布

阅读量843

点赞数

分类专栏： CUDA编程文章标签： P2P通信 GPU CUDA

本文链接：https://blog.csdn.net/u012229282/article/details/124154323

版权

CUDA编程专栏收录该内容

28 篇文章 8 订阅

订阅专栏

本文探讨了如何利用CUDA的P2P通信功能在两个GPU间实现高效的数据拷贝，通过ping-pong复制策略，展示了在循环中使用cudaMemcpy进行双向数据交换的示例。特别强调了P2P通信的单向性，并提到了在代码片段中的应用实例。

摘要由CSDN通过智能技术生成

在GPU之间开启P2P通信功能后，就可以进行拷贝数据了

for (int i=0; i<100; i++)
{
    // With UVA we don't need to specify source and target devices, the
    // runtime figures this out by itself from the pointers
    // Ping-pong copy between GPUs
    if (i % 2 == 0)
    {
        checkCudaErrors(cudaMemcpy(g1, g0, buf_size, cudaMemcpyDefault));
    }else{
        checkCudaErrors(cudaMemcpy(g0, g1, buf_size, cudaMemcpyDefault));
    }
}

注意 P2P通信是单向的