Cuda统一内存简单测试失败

Question

我是一个新手，第一次尝试测试 Cuda6 统一内存。编译并运行后，我期望得到:38、10，但我得到的结果是:5、10。在这种情况下，我做错了什么让我完全愚蠢吗？非常感谢你们!

#include <iostream>
#include <cuda.h>

__global__ void add2(int *a, int *b)
    {
        *a += 33;
    }

int main(){

    int a = 5;
    int b = 10;

    int *p_a, *p_b;

    cudaMallocManaged(&p_a,sizeof(int));
    cudaMallocManaged(&p_b,sizeof(int));

    p_a = &a;
    p_b = &b;

    add2<<<1,1>>>(p_a, p_b);
    cudaDeviceSynchronize();

    std::cout << *p_a << " " << b << std::endl;
    return 0;
}

Answer 1

cudaMallocManaged 创建分配并相应地设置指针，类似于 malloc。如果你然后用

删除分配的指针

p_a=&a;

您的内核将无法工作。阅读 CUDA 6 编程指南中的部分以了解 UM 的工作原理。

你可以试试

*p_a=a;

对于 b 也是如此。